我正在使用“ multiclass.OneVsRestClassifier”和“ cross_validation.StratifiedKFold”。当我对多标签问题进行交叉验证时,它会失败。是否可以对多标签问题scikit-learn执行交叉验证?
我认为问题出在类标签列表Eg([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)
如果您打算使用scikit-learn解决多标签任务,建议您首先使用MultilabelBinarizer将输出转换为标签二进制指示符。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句