sklearn 中的决策树

2020-03-07  本文已影响0人  有机会一起种地OT

sklearn.tree中包含决策树,和极端随机树两类方法。

DecisionTree

sklearn使用CART方法训练决策树。DecisionTreeClassifier和DecisionTreeRegressor 分别是用于分类和回归的决策树方法。
DecisionTreeClassifier常用参数包括

下列参数用来限制决策树的形状规模大小:

对于DecisionTreeRegressor 参数也是类似的,但度量不纯度的 criterion 有所不同,其可选的参数为“mse”,“friedman_mse”,“mae”,默认为“mse”。

决策树学习器通过 fit 方法进行训练,除了样本X和输出y之外,可以输入sample_weight参数指定样本权重(会影响到上面的min_weight_fraction_leaf,min_impurity_decrease计算方法)

predict 用于预测。在DecisionTreeClassifier中还有predict_proba、predict_log_proba方法用于预测各类的概率和对数几率。

在训练后sklearn会自动计算每个特征的重要性。可以通过学习器的属性 feature_importances_ 进行查看。

ExtraTree

sklearn.tree 中还包含ExtraTreeClassifier 和 ExtraTreeRegreoosr。极端随机树与一般的决策树之间的区别在于树生成的方式。在节点上寻求最佳分割(splitter = “best”)时,随机树会为每个随机选取的max_features个特征进行随机分割,再从中选取最小化不纯度的那个。一般决策树则为每个特征进行最小化不纯度的分割后,再选择其中最小化不纯度的。

如果max_features 为1,那么整个树则为完全随机的。要注意,极端随机数只能在集成方法中使用,不可单独哪来作为学习器,因为其随机性不具备指导意义。实际上,sklearn.ensemble中也有 ExtraTreeClassifier 和 ExtraTreeRegreoosr 两个方法,就是组合了多个极端随机书的集成学习器。一般来说,使用的也是sklearn.ensemble中的这两个方法,而不是sklearn.tree中的。

具体的参数和上面的DecisionTree类似。对于集成学习的方法,要对bootstrap 和 oob_score 参数予以注意,确认数据采样的方式。

输出

sklearn.tree中还包含了export_text 和 export_graphviz 两种输出方法。分别用文本和图像信息两种方式输出树结构。

上一篇下一篇

猜你喜欢

热点阅读