fixes #19 : issue where the model never gets trained after server sta…

…rt with empty dataset

src/api.py

@@ -115,7 +115,7 @@ def predict_fmlearn():
 
     data = {}
 
-    if not fml.is_model_trained():
+    if fml._new_recs == math.inf:
         data['response'] = 'Model not trained!'
         return json.dumps(data)
 

src/fmlearn.py

@@ -123,6 +123,9 @@ def predict(self, X_pred):
         # or if the retrain flag is set to true because new data has been loaded
         if self._retain == True or self._new_recs >= self.MAX_NEW_RECORDS:
             self.train()
+
+        if not self.is_model_trained():
+            self.load_data_and_train()
 
         y_pred = self._model.predict(X_pred)