如何利用Python建立决策树模型

2025-07-12 17:52:23

1、输入数据集。importpandasaspddf=pd.DataFrame({'name':缪梨痤刻['Lily','Lucy','Jim','Tom','Anna','Jack','Sam'],'weight':[42,38,78,67,52,80,92],'height':[162,158,169,170,166,175,178],'is_fat':[0,0,1,0,1,0,1]})

如何利用Python建立决策树模型

3、准备训练集,is_fat为目标变量,'weight'和'height'为自变量。X=df.loc[:,['weight','height']]y=df['is_fat']

如何利用Python建立决策树模型

5、利用模型进行预测。y_pred=clf.predict(X)print(y_pred)

如何利用Python建立决策树模型
声明:本网站引用、摘录或转载内容仅供网站访问者交流或参考,不代表本站立场,如存在版权或非法内容,请联系站长删除,联系邮箱:site.kefu@qq.com。
猜你喜欢