diff --git a/docs/source/handlers.rst b/docs/source/handlers.rst index 270083f717..49c84dab28 100644 --- a/docs/source/handlers.rst +++ b/docs/source/handlers.rst @@ -53,6 +53,12 @@ ROC AUC metrics handler :members: +Average Precision metric handler +-------------------------------- +.. autoclass:: AveragePrecision + :members: + + Confusion matrix metrics handler -------------------------------- .. autoclass:: ConfusionMatrix diff --git a/docs/source/metrics.rst b/docs/source/metrics.rst index 616f0fe385..16593a7d0a 100644 --- a/docs/source/metrics.rst +++ b/docs/source/metrics.rst @@ -80,6 +80,13 @@ Metrics .. autoclass:: ROCAUCMetric :members: +`Average Precision` +-------------------------- +.. autofunction:: compute_average_precision + +.. autoclass:: AveragePrecisionMetric + :members: + `Confusion matrix` ------------------ .. autofunction:: get_confusion_matrix