sklearn训练classifier的时候报错Unknown label type

  统计/机器学习 Python    浏览次数:18323        分享
1

我的代码

clf = tree.DecisionTreeClassifier()
clf.fit(X, y)

报错内容

raise ValueError("Unknown label type: %r" % y_type)
ValueError: Unknown label type: 'continuous'

应该是我的y出了问题,但是我看了数据,我的y都是0,1的,应该没有问题,非常困扰

 

有故事的董同学   2018-07-22 07:53



   1个回答 
6

sklearn默认的y的类型必须是整数型,你的y可能是float类型,你转换一下应该就好了

y = np.array(y, dtype=int)


SofaSofa数据科学社区DS面试题库 DS面经

mrhust   2018-07-22 09:44



  相关讨论

sklearn可以用gpu加速吗?

求sklearn.pipeline.Pipeline使用的例子

sklearn错误ImportError: cannot import name 'joblib'

查看sklearn版本

python sklearn模型中random_state参数的意义

sklearn模型当中的verbose是什么意思?

sklearn中除了sgd以外,还有什么可以使用partial_fit方法吗?

sklearn分类模型中预测结果怎么调整概率的阈值

怎么获取sklearn.linear_model.SGDRegressor的回归系数、截距?

sklearn里的LabelEncoder什么用?

  随便看看

如何在numpy array尾部增加一行

机器学习基础

推荐系统里的ALS是什么意思?

推荐开放数据库

特征归一化对K Means有影响吗?