sklearn中的决策树模块及流程

34 篇文章 2 订阅
6 篇文章 0 订阅

sklearn 中的决策树

一、模块sklearn.tree

sklearn中的决策树都在“tree”这个模块之下,这个模块共包括五个类:

tree.DecisionTreeClassifier分类树
tree.DecisionTreeRegressor回归树
tree.export_graphviz将生成的决策树导出为DOT格式,画图专用
tree.ExtraTreeClassifier高随机版本的分类树
tree.ExtraTreeRegressor高随机版本的回归树

二、sklearn的基本建模流程

在这个过程中,分类树对应的代码是:

from sklearn import tree   # 需要导入的模块

clf = tree.DecisionTreeClassifier()   # 实例化

clf = clf.fit(x_train,y_train)   # 用训练集数据训练模型

result = clf.score(x_test,y_test)   # 导入测试集,从接口中调用需要的信息

(根据菜菜的机器学习整理)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值