Update predict.py

McGill-MMA-EnterpriseAnalytics · Apr 26, 2024 · 781ae51 · 781ae51
1 parent 0a129d2
commit 781ae51
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/09_model_pipeline/predict/src/predict.py b/09_model_pipeline/predict/src/predict.py
@@ -1,14 +1,15 @@
 from utils import * 
 import numpy
+import pickle
 class Predict:
     def main_predict(data, bucket_name, path):
-        model = load_models_from_s3(bucket_name,  path )
-
+        #model = load_models_from_s3(bucket_name,  path )
+        model = pickle.load(open("trained_model.pickle",'r'))
         probas = model.predict_proba(data)
         preds = np.where(probas[:,1] >= 0.4, 1, 0)
         data['prob_0'] = probas[:,0]
         data['prob_1'] = probas[:,1]
         data['predictions'] = preds
         record = data.to_dict(orient='records')[0]
         json_result = data.to_json(orient='records', lines=True).splitlines()
-        return json_result
+        return json_result