有没有办法在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）。