Skip to content

Commit 942d045

Browse files
author
Boyan Hristov
committed
#104 - FIXED: now saving transformed data in fit() of base learner
1 parent 171e2e9 commit 942d045

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

modAL/models/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,7 @@ def fit(self, X: modALinput, y: modALinput, bootstrap: bool = False, **fit_kwarg
211211
check_X_y(X, y, accept_sparse=True, ensure_2d=False, allow_nd=True, multi_output=True, dtype=None,
212212
force_all_finite=self.force_all_finite)
213213
self.X_training, self.y_training = X, y
214+
self.Xt_training = self.transform_without_estimating(self.X_training) if self.on_transformed else None
214215
return self._fit_to_known(bootstrap=bootstrap, **fit_kwargs)
215216

216217
def predict(self, X: modALinput, **predict_kwargs) -> Any:

0 commit comments

Comments
 (0)