Custom fitness function for yolo11m-cls hyperparameter tuning

Toxite · February 10, 2025, 11:09pm

You need to modify source code for that.

ultralytics/utils/metrics.py

9d8c2fe3c


      
          def process(self, targets, pred):
              """Target classes and predicted classes."""
              pred, targets = torch.cat(pred), torch.cat(targets)
              correct = (targets[:, None] == pred).float()
              acc = torch.stack((correct[:, 0], correct.max(1).values), dim=1)  # (top1, top5) accuracy
              self.top1, self.top5 = acc.mean(0).tolist()
          
          @property
          def fitness(self):
              """Returns mean of top-1 and top-5 accuracies as fitness score."""
              return (self.top1 + self.top5) / 2
          
          @property
          def results_dict(self):
              """Returns a dictionary with model's performance metrics and fitness score."""
              return dict(zip(self.keys + ["fitness"], [self.top1, self.top5, self.fitness]))
          
          @property
          def keys(self):
              """Returns a list of keys for the results_dict property."""
              return ["metrics/accuracy_top1", "metrics/accuracy_top5"]

Topic		Replies	Views
Model tune for Trained model YOLO yolo , question , support , troubleshooting	9	285	November 16, 2025
Looking for support regarding the custom fitness metric without editing the source code YOLO support	1	140	July 5, 2025
WARNING ❌️ training failure for hyperparameter tuning Support support	4	970	October 5, 2024
Poor training results after tuning yolo11 Discussion yolo , troubleshooting	1	239	October 9, 2025
New Release: Ultralytics v8.3.198 Discussion releases , announcements , ultralytics-official	0	68	September 11, 2025

Custom fitness function for yolo11m-cls hyperparameter tuning

Related topics