Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -64,7 +64,7 @@ def train_the_model(data,page):
|
|
64 |
}
|
65 |
|
66 |
# Initializing GridSearchCV
|
67 |
-
grid_search = GridSearchCV(xgb_model, param_grid, cv=
|
68 |
|
69 |
# Fitting GridSearchCV
|
70 |
grid_search.fit(X_train, y_train)
|
@@ -117,7 +117,7 @@ def train_the_model(data,page):
|
|
117 |
xgb = XGBClassifier(use_label_encoder=False, eval_metric='logloss')
|
118 |
|
119 |
# Setup GridSearchCV
|
120 |
-
grid_search = GridSearchCV(xgb, param_grid, cv=
|
121 |
|
122 |
# Fit the grid search to the data
|
123 |
grid_search.fit(X_train, y_train)
|
|
|
64 |
}
|
65 |
|
66 |
# Initializing GridSearchCV
|
67 |
+
grid_search = GridSearchCV(xgb_model, param_grid, cv=10, n_jobs=-1, scoring='accuracy')
|
68 |
|
69 |
# Fitting GridSearchCV
|
70 |
grid_search.fit(X_train, y_train)
|
|
|
117 |
xgb = XGBClassifier(use_label_encoder=False, eval_metric='logloss')
|
118 |
|
119 |
# Setup GridSearchCV
|
120 |
+
grid_search = GridSearchCV(xgb, param_grid, cv=10, n_jobs=-1, scoring='accuracy')
|
121 |
|
122 |
# Fit the grid search to the data
|
123 |
grid_search.fit(X_train, y_train)
|