有没有办法在 python 中打印一个 XGBoostRegressor 树？

Question

我已经构建了一个 XGBoostRegressor 模型，我现在想尝试绘制其中一棵树。我知道常规的 xgb 分类器有函数 plot_tree 但不幸的是 XGBoostRegressor 没有。有没有其他方法可以绘制树？我还尝试从 xgboost 导入 plot_tree 并使用返回的 plot_tree(xgb)ValueError('Unable to parse node: 44['product_family'])如果有任何其他方法可以做到这一点，有什么想法吗？

Answer 1

我发现了错误，我的一些功能名称中有一些空格。我添加了以下行

    df.columns = df.columns.str.replace(" ", "_")

现在它可以使用 plot_tree(xgb)。