Update main.py

main.py

@@ -64,7 +64,7 @@ def train_the_model(data,page):
         }
         
         # Initializing GridSearchCV
-        grid_search = GridSearchCV(xgb_model, param_grid, cv=1, n_jobs=-1, scoring='accuracy')
+        grid_search = GridSearchCV(xgb_model, param_grid, cv=10, n_jobs=-1, scoring='accuracy')
         
         # Fitting GridSearchCV
         grid_search.fit(X_train, y_train)
@@ -117,7 +117,7 @@ def train_the_model(data,page):
         xgb = XGBClassifier(use_label_encoder=False, eval_metric='logloss')
         
         # Setup GridSearchCV
-        grid_search = GridSearchCV(xgb, param_grid, cv=1, n_jobs=-1, scoring='accuracy')
+        grid_search = GridSearchCV(xgb, param_grid, cv=10, n_jobs=-1, scoring='accuracy')
         
         # Fit the grid search to the data
         grid_search.fit(X_train, y_train)