Skip to content

Commit 48c1986

Browse files
committed
Added the model traniner starting code
1 parent 2c9449d commit 48c1986

File tree

2 files changed

+20
-1
lines changed

2 files changed

+20
-1
lines changed

networksecurity/components/data_validation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

networksecurity/components/model_trainer.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,3 +6,22 @@
66
from networksecurity.entity.config_entity import ModelTrainerConfigEntity , DataTransformationConfigEntity
77
from networksecurity.entity.artifact_entity import DataTransformationArtifactEntity , ModelTrainerArtifactEntity
88
from 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)

0 commit comments

Comments
 (0)