Commit a8b67b39 authored by miaecle's avatar miaecle
Browse files

fix

parent eb302b69
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -73,7 +73,7 @@ def run_benchmark(datasets,
    ]:
      mode = 'regression'
      if metric == None:
        metric = str('r2'
        metric = str('r2')
    else:
      raise ValueError('Dataset not supported')

+1 −1
Original line number Diff line number Diff line
@@ -71,7 +71,7 @@ def run_benchmark_low_data(datasets,
    }

    if isinstance(metric, str):
      metric = [metric_all[metric]]
      metric = metric_all[metric]

    if featurizer == None and isinstance(model, str):
      # Assigning featurizer if not user defined