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(
810810 model = self .get_model ()
811811 hp_fit = self ._get_params_fit ()
812812 model = model .fit (df , ** hp_fit )
813- self .model_last = copy (model )
813+ self ._model_fitted = copy (model )
814814 return model
815815
816816 def _transform_element (
@@ -878,7 +878,7 @@ def get_summary_training(self) -> Dict:
878878 Summary of the training
879879
880880 """
881- model = self .model_last
881+ model = self ._model_fitted
882882 return model .summary
883883
884884 def get_summary_architecture (self ) -> Dict :
@@ -890,7 +890,7 @@ def get_summary_architecture(self) -> Dict:
890890 Summary of the architecture
891891
892892 """
893- model = self .model_last
893+ model = self ._model_fitted
894894 eps_model = model ._get_eps_model ()
895895 return {
896896 "number_parameters" : model .get_num_params (),
You can’t perform that action at this time.
0 commit comments