holisticai.bias.plots.success_rate_curve

holisticai.bias.plots.success_rate_curve(group_a, group_b, y_pred, ax=None, size=None, title=None)

Success rate A vs B curve.

Description

This function plots the group_a pass rate vs group_b pass rate curve.

Parameters

group_aarray-like

Group membership vector (binary)

group_barray-like

Group membership vector (binary)

y_predarray-like

Predictions vector (regression)

ax (optional)matplotlib axes

Pre-existing axes for the plot

size (optional)(int, int)

Size of the figure

title (optional)str

Title of the figure

Returns

matplotlib ax