Hugging Face's logo

Spaces:
Gaykar
/
ClassifyEmail
Running

App Files Community
Gaykar commited on
Commit
64ec4c9
·
1 Parent(s): dda42a2

made pred_prob

Browse files
Files changed (1) hide show
  1. preditormodels.py +1 -1
preditormodels.py CHANGED
@@ -52,7 +52,7 @@ class PhishingPredictor:
52
 
53
 
54
 
55
- prob = self.xgb_model.inplace_predict(final_input)[0][1]
56
 
57
  prediction = "PHISHING" if prob > 0.5 else "SAFE"
58
 
 
52
 
53
 
54
 
55
+ prob = self.xgb_model.predict_proba(final_input)[0][1]
56
 
57
  prediction = "PHISHING" if prob > 0.5 else "SAFE"
58