决策树

决策树的构造与涉及参数

树模型参数:

1.criteration gini or entropy

2.spliter best (默认)or random

3.max features None

4.max_detph

5.min_sample_split(小于某个值的时候不再分裂)

6.min_sample_leaf(叶子节点最小的样本数,小于该数则剪枝)

7.min_weight_fraction_leaf(叶子节点权重和的最小值)

8.max_leaf_nodes

9.class_weight

10.min_impurity_split

n_estimators

import matplotlib.pyplot as pd
import pandas as pd
from sklearn.datasets.california_housing import fetch_california_housing
housing=fetch_california_housing()
print(housing.DESCR)
from sklearn import tree
dtr=tree.DecisionTreeRegressor(max_depth=2)
#传递x,y参数:x,数据;y,label
print(dtr.fit(housing.data[:,[6,7]],housing.target))
#可视化显示需要安装graphviz   http://www.graphviz.org/Download..php
dot_data= \
    tree.export_graphviz(
        dtr,
        out_file=None,
        feature_names=housing.feature_names[6:8],#特征名字
       
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值