Skip to content

Commit 1351dd2

Browse files
committed
Fix X data passed in to loss
1 parent fd0ea02 commit 1351dd2

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

autosklearn/evaluation/train_evaluator.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -763,6 +763,8 @@ def _partial_fit_and_predict_iterative(
763763
if model.estimator_supports_iterative_fit():
764764
X = select(self.X_train, train_indices)
765765
y = select(self.Y_train, train_indices)
766+
767+
X_test = select(self.X_train, test_indices)
766768
y_test = select(self.Y_train, test_indices)
767769

768770
Xt, fit_params = model.fit_transformer(X, y)
@@ -797,7 +799,7 @@ def _partial_fit_and_predict_iterative(
797799
self.model = model
798800

799801
train_loss = self._loss(y, Y_train_pred, X_data=X)
800-
loss = self._loss(y_test, Y_optimization_pred, X_data=X)
802+
loss = self._loss(y_test, Y_optimization_pred, X_data=X_test)
801803

802804
additional_run_info = model.get_additional_run_info()
803805

0 commit comments

Comments
 (0)