决策树分类器sklearn.tree.DecisionTreeClassifier的使用

sklearn.tree.DecisionTreeClassifier(criterion=’gini’, splitter=’best’, max_depth=None, min_samples_split=2, min_samples_leaf=1, min_weight_fraction_leaf=0.0, max_features=None, random_state=None, max_leaf_nodes=None, min_impurity_decrease=0.0, min_impurity_split=None, class_weight=None, presort=False)
常用的参数criterion、splitter(样本数少用best、样本多用random取局部最优点)、max_depth(样本数少、特征少可以不管他,如果特征较多、样本较多要限制一下,10-100视情况而定)。
具体的各参数含义可以参考API文档及其他技术博客。
分类器常用的方法:
fit(self, X, y, sample_weight=None, check_input=True, X_idx_sorted=None),按照x、y建立一个决策树分类器。
predict(self, X, check_input=True),输入x,预测y的类别,返回各个样本的预测类别array或者list,shape为[n_samples]
predict_proba(self, X, check_input=True),输入x,返回各样本的各种预测类别的概率值,array of shape = [n_samples, n_classes], or a list of n_outputs

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值