Skip to content

Commit 17ae35f

Browse files
committed
Added the estimator.py
1 parent 1ce3e36 commit 17ae35f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

networksecurity/components/model_trainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
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+
from networksecurity.utils.model_metric.estimator import NetworkModel
910

1011
class ModelTrainer:
1112
def __init__(self, model_trainer_config: ModelTrainerConfigEntity,

0 commit comments

Comments
 (0)