Skip to content

Commit

Permalink
Merge pull request #59 from McGill-MMA-EnterpriseAnalytics/aladelca-p…
Browse files Browse the repository at this point in the history
…atch-25

Update predict.py
  • Loading branch information
aladelca authored Apr 26, 2024
2 parents a098928 + 413a779 commit 02328e4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion 09_model_pipeline/predict/src/predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
class Predict:
def main_predict(data, bucket_name, path):
#model = load_models_from_s3(bucket_name, path )
model = pickle.load(open("trained_model.pickle",'r'))
model = pickle.load(open("src/trained_model.pickle",'rb'))
probas = model.predict_proba(data)
preds = np.where(probas[:,1] >= 0.4, 1, 0)
data['prob_0'] = probas[:,0]
Expand Down

0 comments on commit 02328e4

Please sign in to comment.