绘制多类[0,1,2] TPOT分类器的ROC曲线时遇到问题
这是我的TPOT python代码,我使用了fi_macro,输出结果可以显示分数。
tpot = TPOTClassifier(generations=5, population_size=10, mutation_rate=0.9,crossover_rate=0.1,verbosity=2, cv=10,scoring='f1_macro')
tpot.fit(X_train, y_train)
print(tpot.score(X_test, y_test))
#to plot the RUC
pred = tpot.predict_proba(Feature_Test)[:,1]
fpr, tpr, thresh = roc_curve(Class_Test, pred)
fpr, tpr, thresh = metrics.roc_curve(Class_Test, pred)
auc = metrics.roc_auc_score(Class_Test, pred)
plt.plot(fpr,tpr,label="size=40, auc="+str(auc))
plt.plot([0, 1], [0, 1], 'k--')
plt.plot(fpr, tpr,ls="None")
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.title('ROC Curve on BreastCancer-Different Population Size')
plt.legend(loc=0)
错误消息提及ValueError:数据不是二进制且未指定pos_label