Arafath10 commited on
Commit
e55794f
1 Parent(s): c27eb57

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -232,6 +232,7 @@ def predict(
232
  # Output
233
  predicted_status = "Unknown" if pred[0] == -1 else encoders['status_name'].inverse_transform([pred])[0]
234
  probability = pred_proba[0][pred[0]] * 100 if pred[0] != -1 else "Unknown"
 
235
 
236
  if predicted_status == "Returned to Client":
237
  probability = 100 - probability
 
232
  # Output
233
  predicted_status = "Unknown" if pred[0] == -1 else encoders['status_name'].inverse_transform([pred])[0]
234
  probability = pred_proba[0][pred[0]] * 100 if pred[0] != -1 else "Unknown"
235
+ print(predicted_status)
236
 
237
  if predicted_status == "Returned to Client":
238
  probability = 100 - probability