You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

scikit-learn决策树代码出现语法错误,请求排查问题原因

代码问题分析及修正
  • 核心语法错误:导入语句错误
    Python中scikit-learn的包名称是sklearn,而非带连字符的scikit-learn,原代码的from scikit-learn import tree会直接触发语法错误——Python不允许导入语句中出现连字符,解析时会将其识别为减法运算符,导致语法逻辑混乱。

  • 修正后的完整代码

from sklearn import tree

# 加载数据
X = [[65,9],[67,7],[70,11],[62,6],[60,7],[72,13],[66,10],[67,7.5]]
Y=["male","female","male","female","female","male","male","female"]

# 构建模型
clf = tree.DecisionTreeClassifier()

# 训练模型
clf.fit(X, Y)

# 预测类别
prediction = clf.predict([[68,9],[66,9]])

# 预测类别概率
probs = clf.predict_proba([[68,9],[66,9]])

# 打印结果
print(prediction)
print(probs)
  • 其他说明
    代码其余部分逻辑正常:使用决策树分类器训练性别预测模型,支持类别预测和概率输出,修正导入语句后即可正常运行。

内容的提问来源于stack exchange,提问作者Freya the Goddess

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.30 12:59:14