小编Roa*_*ach的帖子

如何获取scikit-learn的DecisionTreeRegressor中节点的MSE?

在生成的决策树回归模型中,graphviz用于查看树结构时有一个MSE属性。我需要获取每个叶子节点的MSE,并根据MSE进行后续操作。但是,在阅读文档后,我找不到提供输出 MSE 的方法。其他属性如特征名称、样本数、预测值等,都有对应的方法:

树状结构

使用help(sklearn.tree._tree.Tree),我可以看到大多数属性都有一些输出值的方法,但是我没有看到有关 MSE 的任何内容。

帮助模块 sklearn.tree._tree 中的 Tree 类 帮助模块 sklearn.tree._tree 中的 Tree 类

python scikit-learn

7
推荐指数
1
解决办法
1765
查看次数

标签 统计

python ×1

scikit-learn ×1