Update Classify fitness definition to mean of top1, top5 (#3789)

2023-07-17 14:12:52 +02:00
parent 9b723d7cfa
commit 96cc28b439
1 changed files with 2 additions and 2 deletions
--- a/ultralytics/utils/metrics.py
+++ b/ultralytics/utils/metrics.py
@ -963,8 +963,8 @@ class ClassifyMetrics(SimpleClass):

    @property
    def fitness(self):
-        """Returns top-5 accuracy as fitness score."""
-        return self.top5
+        """Returns mean of top-1 and top-5 accuracies as fitness score."""
+        return (self.top1 + self.top5) / 2

    @property
    def results_dict(self):