analist commited on
Commit
4c41588
·
verified ·
1 Parent(s): 93fbd34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = {}