njab.plotting.metrics module

njab.plotting.metrics module#

njab.plotting.metrics.plot_auc(results: Results, ax: Axes | None = None, label_train='train', label_test='test', **kwargs) Axes[source]#

Plot ROC curve for train and test data.

njab.plotting.metrics.plot_prc(results: ResultsSplit, ax: Axes | None = None, label_train='train', label_test='test', **kwargs)[source]#

Plot precision recall curve for train and test data.

njab.plotting.metrics.plot_split_auc(result: ResultsSplit, name: str, ax: Axes) Axes[source]#

Add receiver operation curve to ax of a split of the data.

njab.plotting.metrics.plot_split_prc(result: ResultsSplit, name: str, ax: Axes) Axes[source]#

Add precision recall curve to ax of a split of the data.