Gourav18 commited on
Commit
52e02ca
·
verified ·
1 Parent(s): aef9b21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -486,9 +486,9 @@ def main():
486
  y_prob = pipeline.predict_proba(X_test)[:, 1] if hasattr(pipeline, "predict_proba") else None
487
 
488
  accuracy = accuracy_score(y_test, y_pred)
489
- precision = precision_score(y_test, y_pred, average='binary')
490
- recall = recall_score(y_test, y_pred, average='binary')
491
- f1 = f1_score(y_test, y_pred, average='binary')
492
  roc_auc = roc_auc_score(y_test, y_prob) if y_prob is not None else None
493
 
494
  results[name] = {
 
486
  y_prob = pipeline.predict_proba(X_test)[:, 1] if hasattr(pipeline, "predict_proba") else None
487
 
488
  accuracy = accuracy_score(y_test, y_pred)
489
+ precision = precision_score(y_test, y_pred)
490
+ recall = recall_score(y_test, y_pred)
491
+ f1 = f1_score(y_test, y_pred)
492
  roc_auc = roc_auc_score(y_test, y_prob) if y_prob is not None else None
493
 
494
  results[name] = {