Skip to content

Commit 4ea5229

Browse files
committed
Fix mean initialization in mrble clustering.
1 parent 997fbe5 commit 4ea5229

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/magnify/identify.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def fit_1d(points, codes, counts, N=100):
169169
# Initialize the Gaussian components.
170170
for i in range(num_codes):
171171
proportions[i] = np.sum(tag_idxs == i) + 1
172-
means[i] = A * code_ratios[i] + p
172+
means[i] = np.median(X_r[tag_idxs == i], axis=0)
173173
if proportions[i] > 1:
174174
covs[i] += np.cov(X_r[tag_idxs == i], rowvar=False)
175175

0 commit comments

Comments
 (0)