Prova 3

metrica_tesi.py

@@ -98,13 +98,14 @@ class MetricaTesi(evaluate.Metric):
     def _compute(self, predictions, references, actions_seen, max_actions_seen=20):
         """Returns the scores"""
         results = dict()
-        for i in range(max_actions_seen):
+        for i in range(max_actions_seen + 1):
             score = 0.0
             support = sum(n == i for n in actions_seen)
             if support != 0:
                 for prediction, reference, n in zip(predictions, references, actions_seen):
-                    if prediction == reference:
-                        score += 1
+                    if n == i:
+                        if prediction == reference:
+                            score += 1
                 score /= support
             results[f"support_{i}"] = support
             results[f"score_{i}"] = score