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就行了。
版权声明:本文为y_hh_原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。