Check if folders exist.

dynamic_feature_and_xgb_training.py

@@ -126,6 +126,8 @@ def main():
     # Create GridSearchCV with k-fold cross-validation
     k_fold = KFold(n_splits=5, shuffle=True, random_state=42)
     grid_search = GridSearchCV(estimator=xgb_model, param_grid=param_grid, cv=k_fold)
+    best_params = grid_search.best_params_
+    best_model = grid_search.best_estimator_
 
     start = datetime.now()
 
@@ -141,7 +143,12 @@ def main():
 
     # Save the trained model
     with open(f'{save_directory_trained_model}/{model_name}', 'wb') as f:
-        pickle.dump(grid_search, f)
+        pickle.dump(best_model, f)
 
+    filename = f'{save_directory_trained_model}/{model_name}_best_parameters.txt'
+    filehandle = open(filename, "a")
+    header = f'best parameter'
+    filehandle.write(header + "\n")
+    filehandle.close()
 
 main()