fix: At david_map_reduce_weighted.py, fixed herding_weights and `ra…

…ndom_weights` in the plots
gchq · Oct 24, 2024 · 07dbc36 · 07dbc36
1 parent a4cbc80
commit 07dbc36
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/examples/david_map_reduce_weighted.py b/examples/david_map_reduce_weighted.py
@@ -227,7 +227,9 @@ def downsample_opencv(image_path: str, downsampling_factor: int) -> np.ndarray:
         -herding_coreset.coreset.data[:, 0],
         c=herding_coreset.coreset.data[:, 2],
         cmap="gray",
-        s=(5.0 * coreset_size * random_weights * downsampling_factor**2).reshape(1, -1),
+        s=(5.0 * coreset_size * herding_weights * downsampling_factor**2).reshape(
+            1, -1
+        ),
         marker="h",
         alpha=0.8,
     )
@@ -241,9 +243,7 @@ def downsample_opencv(image_path: str, downsampling_factor: int) -> np.ndarray:
         random_coreset.coreset.data[:, 1],
         -random_coreset.coreset.data[:, 0],
         c=random_coreset.coreset.data[:, 2],
-        s=(5.0 * coreset_size * herding_weights * downsampling_factor**2).reshape(
-            1, -1
-        ),
+        s=(5.0 * coreset_size * random_weights * downsampling_factor**2).reshape(1, -1),
         cmap="gray",
         marker="h",
         alpha=0.8,