cap score to 1 (#1383)

Co-authored-by: Kartik Pradeepan <[email protected]>
brain-score · Nov 9, 2024 · 846ffa6 · 846ffa6
1 parent 5adf516
commit 846ffa6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/brainscore_vision/benchmarks/lonnqvist2024/benchmark.py b/brainscore_vision/benchmarks/lonnqvist2024/benchmark.py
@@ -44,6 +44,9 @@ def __call__(self, candidate: BrainModel, return_raw_responses: bool = False):
         # Adjust score to ceiling
         ceiling = self.ceiling
         score = raw_score / ceiling
+        # ensure score <= 1.0
+        if score.values > 1:
+            score = Score(np.array(1.))
         score.attrs['raw'] = raw_score
         score.attrs['ceiling'] = ceiling
         if return_raw_responses:

diff --git a/brainscore_vision/benchmarks/malania2007/benchmark.py b/brainscore_vision/benchmarks/malania2007/benchmark.py
@@ -186,6 +186,9 @@ def __call__(self, candidate: BrainModel):
             # Adjust score to ceiling
             ceiling = self.ceiling
             score = raw_score / ceiling
+            # ensure score <= 1.0
+            if score.values > 1:
+                score = Score(np.array(1.))
             score.attrs['error'] = raw_score.error
 
             score.attrs['raw'] = raw_score