fix: maximum 2K samples for animation

encord-team · Mar 1, 2024 · 188a82a · 188a82a
1 parent b8cecdb
commit 188a82a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/clip_eval/plotting/animation.py b/clip_eval/plotting/animation.py
@@ -221,6 +221,11 @@ def build_animation(
     reduced_1 = standardize(reducer.get_reduction(defn_1))
     reduced_2 = rotate_to_target(standardize(reducer.get_reduction(defn_2)), reduced_1)
 
+    if reduced_1.shape[0] > 2_000:
+        selection = np.random.permutation(reduced_1.shape[0])[2_000]
+        reduced_1 = reduced_1[selection]
+        reduced_2 = reduced_2[selection]
+
     return create_embedding_chart(
         reduced_1,
         reduced_2,