holisticai.bias.plots.rmse_bar_plot

holisticai.bias.plots.rmse_bar_plot(p_attr, y_pred, y_true, ax=None, size=None, title=None)

RMSE Bar Plot

Description

This function plots the RMSE for each group as a bar plot.

Parameters

p_attrarray-like

Protected attribute vector

y_predarray-like

Predictions vector (regression)

y_truearray-like

Target 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