gpt4 book ai didi

python - Xgboost绘图树错误: Unable to parse node: 0:[petal

转载 作者:行者123 更新时间:2023-11-30 09:41:59 25 4
gpt4 key购买 nike

我正在尝试使用plot_tree,如tutorial所示

我正在使用 iris 数据集来训练模型,这是我的代码:

from numpy import loadtxt
from xgboost import XGBClassifier
from xgboost import plot_tree
import matplotlib.pyplot as plt
import shap



X,y = shap.datasets.iris()
y = pd.Series([v if v==1 else 0 for v in y])

model = XGBClassifier()
model.fit(X, y)
# plot single tree
plot_tree(model)
plt.show()

从那里,我收到此错误:

ValueError: Unable to parse node: 0:[petal

我不知道在哪里寻找它,因为该模型可以毫无问题地进行训练和预测。

我正在使用 sklearn 版本“0.20.3”

enter image description here

最佳答案

我遇到了同样的问题,它与 graphviz 安装无关。就我而言,问题是 pandas dataFrame 的一些列名称有空格。另请参阅关于 github 的讨论.

当我添加时

df.rename(columns = lambda x: x.replace(' ', '_'), inplace=True)

根据我的预处理,它解决了这个问题。

关于python - Xgboost绘图树错误: Unable to parse node: 0:[petal,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57422912/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com