plot_brier_curve

Note

This page is a reference documentation. It only explains the function signature, and not how to use it. Please refer to the Habitat Guide and Python API guide (v2.0) for usage.

plot_brier_curve(time: ndarray, event: ndarray, survival_probability: ndarray, times: ndarray, *, time_label: str = 'Time')[source]

Brier score of the predicted survival function across follow-up time.

Parameters:

  • time – Observed follow-up durations.

  • event – Event indicators, True for observed events.

  • survival_probability – (n_subjects, n_times) predicted S(t|x).

  • times – The times the probability columns align to.

  • time_label – X-axis label (sanitised).

Returns:

The matplotlib Figure.