gpt4 book ai didi

python - 如何从 Python 中的 ROC 曲线中获取最佳阈值?

转载 作者:太空宇宙 更新时间:2023-11-03 23:59:31 24 4
gpt4 key购买 nike

<分区>

我想使用 Python 从 ROC 曲线中获取最佳阈值。我知道如何使用坐标函数在 R 中执行此操作,但我似乎无法在 Python 中找到类似的函数。

这是我显示 ROC 曲线的方式

def plot_roc_curve(fpr,tpr, thresholds):
plt.figure()
plt.plot(fpr, tpr, color='darkorange', label='ROC curve (area = %0.2f)' % metrics.auc(fpr, tpr))
plt.plot([0, 1], [0, 1], color='navy', linestyle='--')
plt.xlim([0.0, 1.0])
plt.ylim([0.0, 1.05])
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.legend(loc="lower right")

# create the axis of thresholds (scores)
ax2 = plt.gca().twinx()
ax2.plot(fpr, thresholds, markeredgecolor='r',linestyle='dashed', color='r')
ax2.set_ylabel('Threshold',color='r')
ax2.set_ylim([thresholds[-1],thresholds[0]])
ax2.set_xlim([fpr[0],fpr[-1]])

plt.savefig('roc_and_threshold.png')
plt.close()

enter image description here

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com