We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 935b449 commit 7d744f3Copy full SHA for 7d744f3
optuna_integration/sklearn/sklearn.py
@@ -280,7 +280,10 @@ def _cross_validate_with_pruning(
280
if is_classifier(estimator):
281
partial_fit_params = self.fit_params.copy()
282
y = self.y.values if isinstance(self.y, pd.Series) else self.y
283
- classes = np.unique(y)
+ if y is not None:
284
+ classes = np.unique(y)
285
+ else:
286
+ classes = None
287
288
partial_fit_params.setdefault("classes", classes)
289
0 commit comments