如何查看scikit-learn中RandomForestClassifier树模型的所有公共变量?
查看随机森林中树的公共变量方法
clf.estimators_[0].tree_是Cython实现的对象,没有Python的__dict__属性,所以vars()无法生效。可用以下两种方式获取公共变量:
方法1:用dir()函数列出所有属性
执行代码:
dir(clf.estimators_[0].tree_)
输出会包含所有可访问的属性和方法,像max_depth、n_nodes、n_leaves这类公共变量都在其中,可按需筛选。
方法2:直接使用文档中明确的公共属性
sklearn的Tree类有一系列公开的属性可直接访问,常见的包括:
n_nodes:树的总节点数n_leaves:叶子节点数量feature:每个节点用于分裂的特征索引threshold:每个节点的分裂阈值children_left:每个节点的左子节点索引children_right:每个节点的右子节点索引
直接通过estimator.tree_.属性名访问即可,比如clf.estimators_[0].tree_.n_leaves。
内容的提问来源于stack exchange,提问作者Kagaratsch
相关产品推荐
相关产品推荐

