Merge pull request #53 from EdwardGlockner/master

fixing for different os

predict.py

@@ -8,17 +8,19 @@
 from sklearn import preprocessing
 from imblearn.over_sampling import SMOTE
 from sklearn.model_selection import GridSearchCV
+import sys
+import os 
 
 ##########################################################
 ## FIXING PATH
-
-import sys
-sys.path.append('.')
+sys.path.append(str(sys.path[0][:-14]))
+dirname = os.getcwd()
+sys.path.insert(1, os.path.join(dirname, "general_classes"))
 
 ##########################################################
 ## LOCAL PACKAGES
 
-from general_classes import *
+from general_classes import DataPreparation 
 
 ##########################################################
 ## GLOBALS
@@ -80,4 +82,4 @@ def predicting():
 
 if __name__ == "__main__":
     # training()
-    predicting()
+    predicting()