plot_precision_recall_curve

plot_precision_recall_curve(
    performance_data,
    stratified_by=['probability_threshold'],
    size=600,
)

Plots a Precision-Recall curve from pre-computed performance data.

This function is useful when you have already computed the performance metrics and want to generate a Precision-Recall plot directly.

Parameters

Name	Type	Description	Default
performance_data	pl.DataFrame	A Polars DataFrame with the necessary performance metrics, including precision (ppv) and recall (tpr), along with any stratification variables.	required
stratified_by	Sequence[str]	The columns in performance_data used for stratification. Defaults to ["probability_threshold"].	['probability_threshold']
size	int	The width and height of the plot in pixels. Defaults to 600.	600

Returns

Name	Type	Description
	Figure	A Plotly Figure object representing the Precision-Recall curve.