changed the weights for the fitness function
This commit is contained in:
@ -625,7 +625,7 @@ class Metric(SimpleClass):
|
||||
|
||||
def fitness(self):
|
||||
"""Model fitness as a weighted combination of metrics."""
|
||||
w = [0.0, 0.0, 0.1, 0.9] # weights for [P, R, mAP@0.5, mAP@0.5:0.95]
|
||||
w = [0.1, 0.4, 0.5, 0.0] # weights for [P, R, mAP@0.5, mAP@0.5:0.95]
|
||||
return (np.array(self.mean_results()) * w).sum()
|
||||
|
||||
def update(self, results):
|
||||
|
Reference in New Issue
Block a user