Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ def train_models(X_train, y_train, X_test, y_test):
|
|
| 31 |
|
| 32 |
results = {}
|
| 33 |
for name, model in models.items():
|
| 34 |
-
|
| 35 |
|
| 36 |
# Predictions
|
| 37 |
y_train_pred = model.predict(X_train)
|
|
|
|
| 31 |
|
| 32 |
results = {}
|
| 33 |
for name, model in models.items():
|
| 34 |
+
model.fit(X_train, y_train)
|
| 35 |
|
| 36 |
# Predictions
|
| 37 |
y_train_pred = model.predict(X_train)
|