pavanmutha commited on
Commit
6b7b353
·
verified ·
1 Parent(s): 1fdfe63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -190,6 +190,8 @@ def compare_models():
190
  "Precision": precision_score(y, y_pred, average="weighted", zero_division=0),
191
  "Recall": recall_score(y, y_pred, average="weighted", zero_division=0),
192
  }
 
 
193
  wandb.log({f"{name}_{k.replace(' ', '_').lower()}": v for k, v in metrics.items() if isinstance(v, (float, int))})
194
  results.append(metrics)
195
 
 
190
  "Precision": precision_score(y, y_pred, average="weighted", zero_division=0),
191
  "Recall": recall_score(y, y_pred, average="weighted", zero_division=0),
192
  }
193
+ if wandb.run is None:
194
+ wandb.init(project="model_comparison", name="compare_models", reinit=True)
195
  wandb.log({f"{name}_{k.replace(' ', '_').lower()}": v for k, v in metrics.items() if isinstance(v, (float, int))})
196
  results.append(metrics)
197