From ed791d360102309d7ce7f89089a7a5c73968725a Mon Sep 17 00:00:00 2001 From: Justin Bedo Date: Mon, 16 Jan 2023 11:56:11 +1100 Subject: fix length --- bin/cluster.hs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) -- cgit v1.2.3