有没有一种方法可以用LightGBM分类器将树数据作为列表



在随机林类型模型中,通常有一个类似于"估计器";其将所有的树分割作为列表的列表返回。我似乎找不到与lightgbm类似的东西。我能找到的最接近的是lgb.plot_tree,它可以很好地可视化单个树。但我想使用可视化中显示的变量数据。

我如何获取这些数据?

LightGBM中没有完全相同的东西。但是,您可以使用scikit学习估计器的booster_属性的dump_model或trees_to_dataframe方法,即

clf = lgb.LGBMClassifier().fit(X, y)
clf.booster_.dump_model()
clf.booster_.trees_to_dataframe()

最新更新