albin commited on
Commit
0edb6be
·
1 Parent(s): a7ca74e

modify threshold

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -98,7 +98,7 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
98
  data = dataFrame_features[selected_columns].values.reshape(1, -1)
99
 
100
  prediction_proba = model.predict_proba(data)[:, 1]
101
- threshold = 0.7
102
  print("prediction_proba:", prediction_proba)
103
  output = 1 if prediction_proba >= threshold else 0
104
 
 
98
  data = dataFrame_features[selected_columns].values.reshape(1, -1)
99
 
100
  prediction_proba = model.predict_proba(data)[:, 1]
101
+ threshold = 0.9
102
  print("prediction_proba:", prediction_proba)
103
  output = 1 if prediction_proba >= threshold else 0
104