我已經構建了一個 XGBoostRegressor 模型，我現在想嘗試繪制其中一棵樹。我知道常規的 xgb 分類器有函數 plot_tree 但不幸的是 XGBoostRegressor 沒有。有沒有其他方法可以繪制樹？我還嘗試從 xgboost 導入 plot_tree 并使用返回的 plot_tree(xgb)ValueError('Unable to parse node: 44['product_family'])如果有任何其他方法可以做到這一點，有什么想法嗎？
1 回答
米脂
TA貢獻1836條經驗 獲得超3個贊
我發現了錯誤，我的一些功能名稱中有一些空格。我添加了以下行
df.columns = df.columns.str.replace(" ", "_")現在它可以使用 plot_tree(xgb)。
添加回答
舉報
0/150
提交
取消