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)
|