We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fd0ea02 commit 1351dd2Copy full SHA for 1351dd2
1 file changed
autosklearn/evaluation/train_evaluator.py
@@ -763,6 +763,8 @@ def _partial_fit_and_predict_iterative(
763
if model.estimator_supports_iterative_fit():
764
X = select(self.X_train, train_indices)
765
y = select(self.Y_train, train_indices)
766
+
767
+ X_test = select(self.X_train, test_indices)
768
y_test = select(self.Y_train, test_indices)
769
770
Xt, fit_params = model.fit_transformer(X, y)
@@ -797,7 +799,7 @@ def _partial_fit_and_predict_iterative(
797
799
self.model = model
798
800
801
train_loss = self._loss(y, Y_train_pred, X_data=X)
- loss = self._loss(y_test, Y_optimization_pred, X_data=X)
802
+ loss = self._loss(y_test, Y_optimization_pred, X_data=X_test)
803
804
additional_run_info = model.get_additional_run_info()
805
0 commit comments