File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -810,7 +810,7 @@ def _fit_element(
810
810
model = self .get_model ()
811
811
hp_fit = self ._get_params_fit ()
812
812
model = model .fit (df , ** hp_fit )
813
- self .model_last = copy (model )
813
+ self ._model_fitted = copy (model )
814
814
return model
815
815
816
816
def _transform_element (
@@ -878,7 +878,7 @@ def get_summary_training(self) -> Dict:
878
878
Summary of the training
879
879
880
880
"""
881
- model = self .model_last
881
+ model = self ._model_fitted
882
882
return model .summary
883
883
884
884
def get_summary_architecture (self ) -> Dict :
@@ -890,7 +890,7 @@ def get_summary_architecture(self) -> Dict:
890
890
Summary of the architecture
891
891
892
892
"""
893
- model = self .model_last
893
+ model = self ._model_fitted
894
894
eps_model = model ._get_eps_model ()
895
895
return {
896
896
"number_parameters" : model .get_num_params (),
You can’t perform that action at this time.
0 commit comments