Added the config entity artifact entity

Author: Ananddd06

networksecurity/constants/training_pipeline/__init__.py

@@ -61,5 +61,5 @@
 MODEL_TRAINER_TRAINED_MODEL_DIR: str = "trained_model"
 MODEL_TRAINER_TRAINED_MODEL_NAME: str = "model.pkl"
 MODEL_TRAINER_EXPECTED_SCORE: float = 0.6
-MODEL_TRAINER_OVER_FITTING_UNDER_FITTING_THRESHOLD : float = 0.05
+MODEL_TRAINER_OVER_FITTING_UNDER_FITTING_THRESHOLD :float = 0.05
 MODEL_TRAINER_MODEL_CONFIG_FILE_PATH: str = os.path.join("config", "model.yaml")

networksecurity/entity/artifact_entity.py

@@ -20,4 +20,17 @@ class DataTransformationArtifact:
     transformed_object_file_path: str
     transformed_train_file_path: str
     transformed_test_file_path: str
+
+@dataclass
+class ClassificationMetricArtifact:
+    f1_score: float
+    r2_score: float
+    precision: float
+    recall: float
+
+@dataclass
+class ModelTrainerArtifact:
+    trained_model_file_path: str
+    trained_metric_artifact: ClassificationMetricArtifact
+    test_metric_artifact: ClassificationMetricArtifact
 

networksecurity/entity/config_entity.py

@@ -124,7 +124,7 @@ def __post_init__(self):
             training_pipeline.MODEL_TRAINER_TRAINED_MODEL_DIR
         )
         self.expected_score: float = training_pipeline.MODEL_TRAINER_EXPECTED_SCORE
-        self.overfitting_underfitting_threshold: training_pipeline.MODEL_TRAINER_OVER_FITTING_UNDER_FITTING_THRESHOLD# Assuming you want to keep this as a constant
+        self.overfitting_underfitting_threshold: float = training_pipeline.MODEL_TRAINER_OVER_FITTING_UNDER_FITTING_THRESHOLD
         self.model_config_file_path: str = os.path.join(
             self.model_trainer_dir,
             training_pipeline.MODEL_FILE_PATH