Spaces:
Running
Running
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- src/dataloading.py +4 -0
- 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}")
|