ValueError: pos_label=1 不是有效标签：array(['COLLECTION', 'PAIDOFF'], dtype='<U10')

Question

我首先遇到了pos_label这个错误，我首先导入了这个库

from sklearn.metrics import jaccard_similarity_score

因为版本错误我再次导入这个库

from sklearn.metrics import jaccard_score

这是我的运行代码：

depth_range = range(1, 10)
jaccard_similarity_score_ = []
f1_score_ = []

for d in depth_range:
    dt = DecisionTreeClassifier(criterion = 'gini', max_depth = d)
    dt.fit(X_train, y_train)
    dt_yhat = dt.predict(X_test)
    ja = jaccard_score(y_test, dt_yhat, "yes")
    jaccard_similarity_score_.append(ja)
    f1_score_.append(f1_score(y_test, dt_yhat, average = 'weighted'))

我收到此错误：

ValueError: pos_label=1 is not a valid label: array(['COLLECTION', 'PAIDOFF'], dtype='

Answer 1

您需要在要预测为 1 的类别中提及 pos_label 值。因此，对于您的情况，PAIDOFF 将是一个正标签，因此请包括 pos_label='PAIDOFF'。