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 = {}
|