python-3.x - 如何修复第三个打印语句？

Question

我正在尝试打印以下代码行。但是，我收到以下错误：“numpy.ndarray”对象对于第三个打印行没有属性“predict”。来自 sklearn.tree

import DecisionTreeClassifier
from sklearn import metrics
X_trainset, X_testset, y_trainset, y_testset = train_test_split(X,y, test_size = 0.2,random_state = 4)drugTree = DecisionTreeClassifier(criterion = 'entropy', max_depth = 4)
drugTree.fit(X_trainset,y_trainset)predTree = drugTree.predict(X_testset)
print('Test set Accuracy:', metrics.accuracy_score(y_testset, predTree))
print("Train set Accuracy: ", metrics.accuracy_score(ytrainset, predTree.predict(X_trainset)))
print("Jaccard:",jaccard_similarity_score(y_testset,predTree))
print("F1_Score:" ,f1_score(y_testset,predTree,average = 'weighted'))

标签： python-3.x

Answer 1

在不了解其中的任何内容的情况下，纯粹基于文档和名称模式，我猜测当您推出 Train set Accuracy 时，您想要drugTree.predict(X_trainset)而不是predTree.predict(X_trainset).