schwellenwahl.png

def fig_threshold(proba, y_test):
    fpr, tpr, thr = roc_curve(y_test, proba)
    sens, spec = tpr, 1 - fpr
    youden = int(np.argmax(tpr - fpr))
    fig, ax = plt.subplots(figsize=(6, 3.6))
    ax.plot(thr, sens, color=PRIMARY, label="Sensitivität")
    ax.plot(thr, spec, color=EVENT, label="Spezifität")
    ax.axvline(thr[youden], color=SECONDARY, ls="--", lw=1)
    ax.text(thr[youden], 0.05, f" Youden = {thr[youden]:.2f}", color=SECONDARY, fontsize=9)
    ax.set_xlim(0, 1)
    ax.set_xlabel("Entscheidungsschwelle")
    ax.set_ylabel("Anteil")
    ax.set_title("Die Schwelle bestimmt Sensitivität und Spezifität")
    ax.legend(loc="center right")
    save(fig, ASSETS / "schwellenwahl.png")