Skip to content

Commit 9dde1aa

Browse files
Julien RousselJulien Roussel
authored andcommitted
public attribute renamed
1 parent 37564b4 commit 9dde1aa

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

qolmat/imputations/imputers_pytorch.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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(),

0 commit comments

Comments
 (0)