决策树CART算法分类性别:解决ValueError: 'continuous'错误
决策树CART算法分类性别:解决'ValueError: Unknown label type: 'continuous''错误
本文将介绍使用决策树CART算法根据以下特征来分类性别时遇到的'ValueError: Unknown label type: 'continuous''错误,并提供了解决方法:
- long_hair 整型 是否长短发
- forehead_width_cm 浮点型 前额宽度
- forehead_height_cm 浮点型 前额高度
- nose_wide 整型 是否鼻宽
- nose_long 整型 是否鼻长
- lips_thin 整型 是否嘴唇薄
- distance_nose_to_lip_long 整型 鼻子到嘴唇距离是否长
错误原因:
这个错误是因为你的y标签(性别)是连续的浮点数值,而决策树算法需要的是离散的类别标签。
解决方法:
将y标签转换成离散的类别标签,比如将男性标记为1,女性标记为0。可以使用sklearn中的LabelEncoder进行转换。
代码示例:
from sklearn.preprocessing import LabelEncoder
# 加载数据集,省略部分代码
# ...
# 将y标签转换成离散的类别标签
y = LabelEncoder().fit_transform(y)
# 训练决策树模型,省略部分代码
# ...
这样就可以避免报错了。
原文地址: https://www.cveoy.top/t/topic/op9l 著作权归作者所有。请勿转载和采集!