Fix make_archetypal_dataset

aleixalcacer · Nov 21, 2023 · ecdfabd · ecdfabd
1 parent e533260
commit ecdfabd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/archetypes/datasets/make_archetypal_dataset.py b/archetypes/datasets/make_archetypal_dataset.py
@@ -67,7 +67,11 @@ def make_archetypal_dataset(
             else:
                 alpha_i = [alpha] * A_i.shape[1]
                 alpha_i[l_i] = 1
-                A_i[i, :] = generator.dirichlet(alpha_i)
+                d_i = -1
+                while d_i != l_i:
+                    d = generator.dirichlet(alpha_i)
+                    d_i = np.argmax(d)
+                A_i[i, :] = d
 
     X = einsum(A, archetypes)