From 699f4f6546d7beaf782b1464aaafbe00c40cf0e4 Mon Sep 17 00:00:00 2001 From: andis Date: Fri, 25 Aug 2017 19:18:36 +0200 Subject: [PATCH] small chang2 --- classifier/__init__.py | 4 +++- storage/__init__.py | 7 +++++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/classifier/__init__.py b/classifier/__init__.py index cf5c18a..e53691d 100644 --- a/classifier/__init__.py +++ b/classifier/__init__.py @@ -1,5 +1,7 @@ from classifier import in_training, print_answers from classifier import get_pipe, test_pipe, get_training_threads -from training import train_single_thread from classifier import store_training_data + +from training import train_single_thread + from prediction import predict_threads diff --git a/storage/__init__.py b/storage/__init__.py index 2a9def7..e6ffa63 100644 --- a/storage/__init__.py +++ b/storage/__init__.py @@ -1,5 +1,8 @@ +# SQL Database session from database import db_session, init_db -from mail_model import Mail -from thread_model import MailThread from database import Base metadata=Base.metadata + +# Two main objects: Mail & MailThread +from mail_model import Mail +from thread_model import MailThread