多标签分类的交叉验证错误

2024-09-30 20:30:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我用的是“multiclass.onevsrestcrifier”和“cross_validation.StratifiedKFold”。当我对多标签问题进行交叉验证时,它失败了。 是否可以对多标签问题scikit learn执行交叉验证?

我认为问题出在类标签列表的元组中,例如([1]、[2]、[2]、[1]、[1,2]、[3]、[1,2,3]…)

我认为此错误的代码如下:

n_samples = X.shape[0]
Y_list = [value for value in Y.T]
print 'Y_list[0].shape:', Y_list[0].shape, 'len(Y_list):', len(Y_list)
cv = cross_validation.StratifiedKFold(Y_list, 3)

Tags: 列表lenvalue标签scikitlearn交叉list