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