We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1ce3e36 commit 17ae35fCopy full SHA for 17ae35f
networksecurity/components/model_trainer.py
@@ -6,6 +6,7 @@
6
from networksecurity.entity.config_entity import ModelTrainerConfigEntity , DataTransformationConfigEntity
7
from networksecurity.entity.artifact_entity import DataTransformationArtifactEntity , ModelTrainerArtifactEntity
8
from networksecurity.utils.main_utils import save_obj , load_obj , load_numpy_array_data
9
+from networksecurity.utils.model_metric.estimator import NetworkModel
10
11
class ModelTrainer:
12
def __init__(self, model_trainer_config: ModelTrainerConfigEntity,
0 commit comments