add granular metrics for accuracy

Author: ashryaagr

sources/full_pipeline.py

@@ -63,7 +63,6 @@ def run_pipeline(file):
     #----------------------------------------------------- Stage 5
 
     result_array = tests.get_evaluation_result_as_numpy()
-    print(result_array)
     from sources.metrics.accuracy import Accuracy
     metric = Accuracy()
     metric.get_metric_value(result_array)

sources/metrics/accuracy.py

@@ -13,6 +13,8 @@ def __init__(self, metric_name="", threshold=None):
 
 
         self.metric_result = None
+        self.metric_result_question_wise = []
+        self.metric_result_question_test_wise = []
         self.threshold = None
 
 
@@ -25,4 +27,6 @@ def passed(self):
     def get_metric_value(self, result_array):
 
         self.metric_result = result_array.mean()
+        self.metric_result_question_wise = result_array.mean(axis=(1, 2, 3)).tolist()
+        self.metric_result_question_test_wise = result_array.mean(axis=(1, 2)).tolist()
         return self.metric_result