seismometer.api.plot_cohort_evaluation

seismometer.api.plot_cohort_evaluation(cohort_col, subgroups, target_column, score_column, thresholds, per_context=False)

Plots model performance metrics split by on a cohort attribute.

Parameters:

• cohort_col (str) – cohort column name

• subgroups (list[str]) – subgroups of interest in the cohort column

• target_column (str) – target column

• score_column (str) – score column

• thresholds (list[float]) – thresholds to highlight

• per_context (bool, optional) – if scores should be grouped, by default False

Returns:

an html visualization of the model performance metrics

Return type:

HTML