Update Classify fitness
definition to mean of top1, top5 (#3789)
This commit is contained in:
@ -963,8 +963,8 @@ class ClassifyMetrics(SimpleClass):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def fitness(self):
|
def fitness(self):
|
||||||
"""Returns top-5 accuracy as fitness score."""
|
"""Returns mean of top-1 and top-5 accuracies as fitness score."""
|
||||||
return self.top5
|
return (self.top1 + self.top5) / 2
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def results_dict(self):
|
def results_dict(self):
|
||||||
|
Reference in New Issue
Block a user