Xawirses/GitMining
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fixed first level of frequent patterns not being filtered

Browse Source
This commit is contained in:
IGI-111
2015-04-03 23:10:11 +02:00
parent 614a615a12
commit 5ba00085d1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Apriori.hs
View File

@@ -52,5 +52,5 @@ singletons table = Set.toList $ foldr (\(ItemSet row) old -> old `Set.union` row
frequentPatterns :: Frequency -> [ItemSet] -> [[ItemSet]] frequentPatterns :: Frequency -> [ItemSet] -> [[ItemSet]]
frequentPatterns thresh table = until (\x -> [] == head x) (\x -> filterByFrequency (generateNextLevel (head x)) : x) [firstLevel] where frequentPatterns thresh table = until (\x -> [] == head x) (\x -> filterByFrequency (generateNextLevel (head x)) : x) [firstLevel] where
firstLevel = map (\x -> ItemSet $ Set.fromList [x]) (singletons table) firstLevel = filterByFrequency $ map (\x -> ItemSet $ Set.fromList [x]) (singletons table)
filterByFrequency = filter (\x -> frequency table x >= thresh) filterByFrequency = filter (\x -> frequency table x >= thresh)