return ranking instead of self-influence values

dilyabareeva · Jun 13, 2024 · 3649eb8 · 3649eb8
1 parent e5682f0
commit 3649eb8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/explainers/aggregators/self_influence.py b/src/explainers/aggregators/self_influence.py
@@ -29,3 +29,4 @@ def get_self_influence_ranking(
             train_ids=[i],
             **explain_fn_kwargs,
         )
+    return self_inf.argsort()