diff options
Diffstat (limited to 'bin/cluster.hs')
-rw-r--r-- | bin/cluster.hs | 2 |
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) |