Commit cd22db00 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #698 from galenxing/add_n_classes

pass n_classes arg to metric.compute_metric
parents e279904f c3cefe3b
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -235,11 +235,11 @@ class GeneratorEvaluator(object):
    for metric in metrics:
      if per_task_metrics:
        multitask_scores[metric.name], computed_metrics = metric.compute_metric(
            y, y_pred, w, per_task_metrics=True)
            y, y_pred, w, per_task_metrics=True, n_classes=self.n_classes)
        all_task_scores[metric.name] = computed_metrics
      else:
        multitask_scores[metric.name] = metric.compute_metric(
            y, y_pred, w, per_task_metrics=False)
            y, y_pred, w, per_task_metrics=False, n_classes=self.n_classes)

    if not per_task_metrics:
      return multitask_scores