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