ValueError: Supported target types are: (‘binary’, ‘multiclass’). Got ‘multilabel-indicator’ instead.
在使用StratifiedKFold进行交叉检验时报错。
kfold
= StratifiedKFold
(n_splits
=10, shuffle
=True, random_state
=42)
for train_index
, test_index
in kfold
.split
(x
, y
):
....
在split函数说明中提到,输入y应该是一个一维的向量,但我进行的是一个多分类,y的数据类型是多维的,因此直接输入的y, 会报错。而且在文档中指出,这个参数默认是None的,说明通常在进行划分数据集的时候是不需要的,因此不传参数y就行了。