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
相关产品推荐
相关产品推荐

