Remove comments

dice-group · Jan 22, 2025 · 7cad63f · 7cad63f
1 parent c1d955c
commit 7cad63f
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/dicee/evaluator.py b/dicee/evaluator.py
@@ -313,12 +313,8 @@ def evaluate_lp_k_vs_all(self, model, triple_idx, info=None, form_of_labelling=N
                 # (5) Sort predictions.
                 sort_values, sort_idxs = torch.sort(predictions, dim=1, descending=True)
 
-                #probabilities = torch.sigmoid(sort_values)
-                #non_zero_probs = probabilities[probabilities != 0.0]
-                #one_minus_non_zero_probs = 1 - non_zero_probs
-
-                non_zero_probs = sort_values[sort_values != 0.0]
-                probabilities = torch.sigmoid(non_zero_probs)
+                non_zero_values = sort_values[sort_values != 0.0]
+                probabilities = torch.sigmoid(non_zero_values)
                 one_minus_non_zero_probs = 1 - probabilities
 
                 ECE = one_minus_non_zero_probs.sum().item()