add average parameter to MeanAveragePrecision to specify micro or mac…

…ro calculation
microsoft · Nov 14, 2023 · 58c16bf · 58c16bf
1 parent 4f3d90c
commit 58c16bf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/responsibleai_vision/responsibleai_vision/rai_vision_insights/rai_vision_insights.py b/responsibleai_vision/responsibleai_vision/rai_vision_insights/rai_vision_insights.py
@@ -1191,6 +1191,7 @@ def compute_object_detection_metrics(
                 continue
 
             metric_OD = MeanAveragePrecision(
+                average=aggregate_method,
                 class_metrics=True,
                 iou_thresholds=normalized_iou_threshold).to(device)
             true_y_cohort = [true_y[cohort_index] for cohort_index