seismometer.api.plot_binary_classifier_metrics

seismometer.api.plot_binary_classifier_metrics(metric_generator, metrics, cohort_dict, target, score_column, *, per_context=False, table_only=False)

Generates a plot with model metrics.

Parameters:

• metric_generator (BinaryClassifierMetricGenerator) – class that creates metrics for a model

• metrics (str | list[str]) – subset of metrics to display

• cohort_dict (dict[str, tuple[Any]]) – dictionary of cohort columns and values used to subselect a population for evaluation

• target (str) – name of the target

• score_column (str) – score column

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

• table_only (bool, optional) – if only the table should be displayed, by default False

Returns:

an html visualization of the model evaluation metrics

Return type:

HTML