Update initialize_system.py

initialize_system.py

@@ -386,7 +386,13 @@ def run_initial_training():
             "data_path": str(dataset_path),
             "class_distribution": class_counts.to_dict(),
             "pipeline_created": pipeline_path.exists(),
-            "individual_components_created": model_path.exists() and vectorizer_path.exists()
+            "individual_components_created": model_path.exists() and vectorizer_path.exists(),
+            "cv_f1_mean": float(cv_results['test_f1_weighted'].mean()),
+            "cv_f1_std": float(cv_results['test_f1_weighted'].std()),
+            "cv_accuracy_mean": float(cv_results['test_accuracy'].mean()),
+            "cv_accuracy_std": float(cv_results['test_accuracy'].std()),
+            "overfitting_score": cv_data['performance_indicators']['overfitting_score'],
+            "stability_score": cv_data['performance_indicators']['stability_score']
         }
 
         metadata_path = path_manager.get_metadata_path()