aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--bin/cluster.hs2
1 files changed, 1 insertions, 1 deletions
diff --git a/bin/cluster.hs b/bin/cluster.hs
index 6bcac29..7984bfd 100644
--- a/bin/cluster.hs
+++ b/bin/cluster.hs
@@ -68,7 +68,7 @@ model xs = do
mapM_ scoreLog $ zipWith likelihood params xs
let cls = take (length xs) clusters
k = maximum cls + 1
- n = length $ head xs
+ n = length (head xs) `div` 2
pure (map (take k) $ take n ps, cls)
where
likelihood ps cnts = product $ zipWith go ps (pairs cnts)