google colab中的dtreeviz类型错误
我想用鸢尾花数据集来制作一个树状图,并展示预测的路径,但我一直遇到一个问题:看看这个
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 个回答
暂无回答