We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b919396 commit cee3313Copy full SHA for cee3313
1 file changed
khiops/sklearn/dataset.py
@@ -614,9 +614,9 @@ def _init_target_column(self, y):
614
else:
615
y_checked = _column_or_1d_with_dtype(y, dtype=y.dtype)
616
elif hasattr(y, "dtypes"):
617
- if isinstance(y.dtypes[0], pd.CategoricalDtype):
+ if isinstance(y.dtypes.iloc[0], pd.CategoricalDtype):
618
y_checked = _column_or_1d_with_dtype(
619
- y, dtype=y.dtypes[0].categories.dtype
+ y, dtype=y.dtypes.iloc[0].categories.dtype
620
)
621
622
y_checked = _column_or_1d_with_dtype(y)
0 commit comments