Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -186,7 +186,7 @@ def train_model(file, n_estimators, learning_rate, max_depth, subsample, progres
|
|
| 186 |
# Evaluate
|
| 187 |
y_pred = model.predict(X_test)
|
| 188 |
accuracy = accuracy_score(y_test, y_pred)
|
| 189 |
-
report = classification_report(y_test, y_pred,
|
| 190 |
accuracies[target_name] = accuracy
|
| 191 |
|
| 192 |
results.append(f"**{target_name} Results**\n")
|
|
|
|
| 186 |
# Evaluate
|
| 187 |
y_pred = model.predict(X_test)
|
| 188 |
accuracy = accuracy_score(y_test, y_pred)
|
| 189 |
+
report = classification_report(y_test, y_pred, zero_division=0)
|
| 190 |
accuracies[target_name] = accuracy
|
| 191 |
|
| 192 |
results.append(f"**{target_name} Results**\n")
|