Commit 32d2a4e1 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Changes

parent b9f19b3d
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -670,6 +670,7 @@ class Metric(object):
      )
    else:
      self.compute_energy_metric = False

    self.metric = metric
    if task_averager is None:
      self.task_averager = np.mean
@@ -688,6 +689,7 @@ class Metric(object):
          self.name = "unknown metric"
    else:
      self.name = name

    if mode is None:
      # These are some smart defaults
      if self.metric.__name__ in [
+0 −2
Original line number Diff line number Diff line
@@ -34,8 +34,6 @@ def test_one_sample():
  ]
  for metric in all_metrics:
    score = metric.compute_singletask_metric(y_true, y_pred, w)
    print("score")
    print(score)


def test_r2_score():