WIP: updated the app.py with fmlearn class test call

mukeshmk · Jun 17, 2020 · fe73f9d · fe73f9d
1 parent 446d426
commit fe73f9d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/app.py b/app.py
@@ -40,7 +40,11 @@ def apidescription():
 # Run Server
 if __name__ == '__main__':
     # app.run()
+
     # NOTE: temp work around so that the server doesn't have to started to test
     # the FMLearn algo. Need to remove once done with it.
-    from src.fmlearn import *
-    print(kmc())
+    from src.fmlearn import fmlearn
+
+    fml = fmlearn()
+    fml._test()
+
diff --git a/src/fmlearn.py b/src/fmlearn.py
@@ -99,9 +99,6 @@ def _test(self, print_details=False):
             # kneighbors()[0] contains distances to points
             # kneighbors()[1] contains indcies of nearest points
             print(model.kneighbors(X_test)[1])
-            
+
             print(y_test.to_string(header=False))
             print(pd.DataFrame(y_pred).to_string(header=False))
-
-if __name__ == "__main__":
-    kmc()