def __init__(self, name, mode, metrics_args):
self.name = name
self.mode = mode
self.num_classes = metrics_args['MODEL']['num_classes']
self.topk = metrics_args['MODEL']['topk']
self.calculator = youtube8m_metrics.EvaluationMetrics(
self.num_classes, self.topk)
def __init__(self, name, phase, **metrics_args):
self.name = name
self.phase = phase
self.metrics_args = metrics_args
self.num_classes = metrics_args['num_classes']
self.topk = metrics_args['topk']
self.calculator = youtube8m_metrics.EvaluationMetrics(
self.num_classes, self.topk)