google colab中的dtreeviz类型错误

0 投票
0 回答
16 浏览
提问于 2025-04-12 07:25

我想用鸢尾花数据集来制作一个树状图,并展示预测的路径,但我一直遇到一个问题:看看这个

TypeError: model() got an unexpected keyword argument 'X'

我尝试运行一些代码:

clas = DecisionTreeClassifier()
iris = load_iris()

X_train = iris.data
y_train = iris.target
clas.fit(X_train,y_train)

x = iris.data[np.random.randint(0,len(iris.data)),:]

viz = dtreeviz.model(clas,
                      X_train=X_train, y_train=y_train,
                      feature_names=iris.feature_names,
                      target_name='iris',
                      class_names=iris.target_names,
                      X=x,
                      show_path=True)
viz.view()

但是总是出现错误,我无法解决这个错误:

TypeError                                 Traceback (most recent call last)
<ipython-input-18-f6761be97782> in <cell line: 10>()
      8 x = iris.data[np.random.randint(0,len(iris.data)),:]
      9 
---> 10 viz = dtreeviz.model(clas,
     11                       X_train=X_train, y_train=y_train,
     12                       feature_names=iris.feature_names,

TypeError: model() got an unexpected keyword argument 'X'

0 个回答

暂无回答

撰写回答