如何解决AttributeError: 'XGBClassifier' 对象没有属性 'plot_tree'
我正在尝试绘制 XGBoost 的第一个决策树,但出现以下错误
AttributeError: 'XGBClassifier' object has no attribute 'plot_tree'
这就是我试图绘制树的方式
import pandas as pd
from xgboost import XGBClassifier,plot_tree
from matplotlib import pyplot as plt
df_train = pd.read_csv("File.csv ")
y = df_train.loc[:,'CLASS']
X= df_train.loc[:,'s1': 's320']
model = XGBClassifier(
scale_pos_weight=1,learning_rate=0.009,colsample_bytree=0.4,subsample=0.8,objective='binary:logistic',n_estimators=1100,reg_alpha=0.3,max_depth=4,gamma=0,use_label_encoder=False)
model.fit(X,y)
plot_tree(model)
plt.show()
我在论坛里没有发现类似的问题,有人知道如何解决吗?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。