File tree Expand file tree Collapse file tree 2 files changed +20
-1
lines changed
networksecurity/components Expand file tree Collapse file tree 2 files changed +20
-1
lines changed Original file line number Diff line number Diff line change @@ -189,4 +189,4 @@ def initiate_data_validation(self) -> DataValidationArtifact:
189189 )
190190
191191 except Exception as e :
192- raise CustomException (e , sys ) from e
192+ raise CustomException (e , sys ) from e
Original file line number Diff line number Diff line change 66from networksecurity .entity .config_entity import ModelTrainerConfigEntity , DataTransformationConfigEntity
77from networksecurity .entity .artifact_entity import DataTransformationArtifactEntity , ModelTrainerArtifactEntity
88from networksecurity .utils .main_utils import save_obj , load_obj , load_numpy_array_data
9+
10+ class ModelTrainer :
11+ def __init__ (self , model_trainer_config : ModelTrainerConfigEntity ,
12+ data_transformation_config : DataTransformationConfigEntity ,
13+ data_transformation_artifact : DataTransformationArtifactEntity ):
14+ try :
15+ self .model_trainer_config = model_trainer_config
16+ self .data_transformation_config = data_transformation_config
17+ self .data_transformation_artifact = data_transformation_artifact
18+ self .logger = Custom_Logger ().get_logger ()
19+ self .logger .info ("Model Trainer initialized with configuration and artifacts." )
20+ except Exception as e :
21+ raise CustomException (e , sys ) from e
22+
23+ def initiate_model_trainer (self ) -> ModelTrainerArtifactEntity :
24+ try :
25+ pass
26+ except Exception as e :
27+ raise CustomException (e , sys )
You can’t perform that action at this time.
0 commit comments