Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,10 +27,10 @@ def train_models(X_train, y_train, X_test, y_test):
|
|
| 27 |
models = {
|
| 28 |
"Logistic Regression": LogisticRegression(random_state=42),
|
| 29 |
"Decision Tree": DecisionTreeClassifier(random_state=42),
|
| 30 |
-
"Random Forest": RandomForestClassifier(random_state=42),
|
| 31 |
"Gradient Boost": GradientBoostingClassifier(random_state=42),
|
| 32 |
-
"Extreme Gradient Boosting": XGBClassifier(random_state=42),
|
| 33 |
-
"Light Gradient Boosting": LGBMClassifier(random_state=42)
|
| 34 |
}
|
| 35 |
|
| 36 |
results = {}
|
|
|
|
| 27 |
models = {
|
| 28 |
"Logistic Regression": LogisticRegression(random_state=42),
|
| 29 |
"Decision Tree": DecisionTreeClassifier(random_state=42),
|
| 30 |
+
"Random Forest": RandomForestClassifier(n_estimators=100, min_samples_split=2,max_features=7, max_depth=None, random_state=42),
|
| 31 |
"Gradient Boost": GradientBoostingClassifier(random_state=42),
|
| 32 |
+
"Extreme Gradient Boosting": XGBClassifier(random_state=42, n_estimators=500, learning_rate=0.0789),
|
| 33 |
+
"Light Gradient Boosting": LGBMClassifier(random_state=42, n_estimators=500, learning_rate=0.0789)
|
| 34 |
}
|
| 35 |
|
| 36 |
results = {}
|