Create artifact_entity.py

src/entity/artifact_entity.py

@@ -0,0 +1,33 @@
+from dataclasses import dataclass
+
+@dataclass
+class DataIngestionArtifact:
+    trained_file_path:str
+    test_file_path:str
+
+@dataclass
+class DataValidationArtifact:
+    validation_status: bool
+    valid_train_file_path: str
+    valid_test_file_path: str
+    invalid_train_file_path: str
+    invalid_test_file_path: str
+    drift_report_file_path: str
+
+@dataclass 
+class DataTransformationArtifact:
+    transformed_object_file_path:str
+    transformed_train_file_path:str 
+    transformed_test_file_path:str 
+
+@dataclass
+class RegressionMetricArtifact:
+    mean_absolute_error:float
+    r2_score:float
+    mean_squared_error:float
+
+@dataclass
+class ModelTrainerArtifact:
+    trained_model_file_path:str
+    train_metric_artifact:RegressionMetricArtifact
+    test_metric_artifact:RegressionMetricArtifact