在用于建模问题的流行软件包中,有许多功能可以将模型对象转换为人类可以理解的形式,例如xgboost中的xgb.model.dt.tree或GBM中的pretty.gbm.tree。catboost包中是否有类似的功能,或者是否有其他方法可以将模型表示为数据框等形式?catboost.Model类的对象是一个包含外部指针和原始数据元素的列表,了解模型结构的唯一方法是将其保存为.py文件,但这是一种相当粗糙的方法。
回答:
虽然有一个tree_idx
属性,但它只能用于与plot_tree
一起进行树的可视化,而不能直接与model
一起使用;这里是一个使用波士顿数据的可复现示例:
model.plot_tree(tree_idx=4)