sklearn 分类器 - predict_proba 最大化 auc 的阈值

Question

我有一个三class class化问题。我训练 classifier 然后绘制不同 classes 的 ROC。

我需要为每个 class 获取最大化 TPR 和最小化 FPR 的阈值。在 Matlab 中，这是返回的。有没有办法在 python / sklearn 中检索它？

谢谢。

Answer 1

所以我的想法是这样的：

import numpy as np
idx = np.linalg.norm(
    (np.array([[0, 1]]) -np.stack([fpr, tpr], axis=1)), 
    axis=1).argmax()
max_thresh = thresholds[idx]

这里做了什么：将 FPR 和 TPR 堆叠在一起作为一个 2D 向量。减去每行向量的左上角并取范数。这将计算距离。拿argmax就知道最大值出现在哪一行。最后，return 该索引的阈值

sklearn classifier - predict_proba threshold that maximizes auc