search

gsganden / model_inspector

A uniform interface to a curated set of methods for inspecting machine learning models
https://gsganden.github.io/model_inspector/
Apache License 2.0
4 stars 0 forks source link

Support PR curves #50

Open gsganden opened 1 year ago

gsganden commented 1 year ago

We can build on calculate_metrics_by_thresh:

results = calculate_metrics_by_thresh_binary(
    y_true=y_true,
    y_prob=y_prob,
    metrics=[metrics.recall_score, metrics.precision_score],
)
ax = results[:-1].plot(x="recall_score", y="precision_score")