Joschka Strueber commited on
Commit
64b132e
·
1 Parent(s): cb7e104

[Ref, Fix] use cached list of usable models, convert logits to OneHot for EC as well

Browse files
Files changed (2) hide show
  1. src/dataloading.py +4 -0
  2. src/similarity.py +1 -0
src/dataloading.py CHANGED
@@ -137,3 +137,7 @@ def load_run_data(model_name, dataset_name):
137
  def load_run_data_cached(model_name, dataset_name):
138
  return load_run_data(model_name, dataset_name)
139
 
 
 
 
 
 
137
  def load_run_data_cached(model_name, dataset_name):
138
  return load_run_data(model_name, dataset_name)
139
 
140
+
141
+ if __name__ == "__main__":
142
+ get_leaderboard_models_reload()
143
+
src/similarity.py CHANGED
@@ -38,6 +38,7 @@ def compute_pairwise_similarities(metric_name: str, probs: list[list[np.array]],
38
  # Convert probabilities to one-hot
39
  probs = [[one_hot(p) for p in model_probs] for model_probs in probs]
40
  elif metric_name == "Error Consistency":
 
41
  metric = EC()
42
  else:
43
  raise ValueError(f"Invalid metric: {metric_name}")
 
38
  # Convert probabilities to one-hot
39
  probs = [[one_hot(p) for p in model_probs] for model_probs in probs]
40
  elif metric_name == "Error Consistency":
41
+ probs = [[one_hot(p) for p in model_probs] for model_probs in probs]
42
  metric = EC()
43
  else:
44
  raise ValueError(f"Invalid metric: {metric_name}")