linted files

This commit is contained in:
IGI-111
2015-04-08 13:39:59 +00:00
parent ae5b95c69d
commit ddffd9a8f1
4 changed files with 16 additions and 14 deletions

View File

@@ -11,7 +11,8 @@ semiUnion :: ItemSet -> ItemSet -> ItemSet
semiUnion (ItemSet set1) (ItemSet set2) = ItemSet $
if max1 <= max2 && Set.delete max1 set1 == Set.delete max2 set2
then set1 `Set.union` set2
else Set.empty where
else Set.empty
where
max1 = Set.findMax set1
max2 = Set.findMax set2
@@ -24,18 +25,21 @@ semiUnion (ItemSet set1) (ItemSet set2) = ItemSet $
-- generate the next level in a bottom-up route
generateNextLevel :: [ItemSet] -> [ItemSet]
generateNextLevel level = trace ("Computing level " ++ show (isSize (head level))) $
foldr (\value old -> generate value ++ old) [] level where
foldr (\value old -> generate value ++ old) [] level
where
generate value = takeWhile (/= empty) $
parMap rpar (semiUnion value) (tail $ List.dropWhile (/= value) level) -- FIXME: this could be a better strategy
isSize (ItemSet set) = Set.size set
singletons :: [ItemSet] -> [Item]
singletons table = Set.toList $ foldr union (Set.fromList []) table where
singletons table = Set.toList $ foldr union (Set.fromList []) table
where
union (ItemSet row) old = old `Set.union` row
frequentPatterns :: Frequency -> [ItemSet] -> [[ItemSet]]
frequentPatterns thresh table = until (\x -> [] == head x)
(\x -> filterByFrequency (generateNextLevel (head x)) : x) [firstLevel] where
(\x -> filterByFrequency (generateNextLevel (head x)) : x) [firstLevel]
where
firstLevel = filterByFrequency $ map (\x -> ItemSet $ Set.fromList [x]) $
trace "Generated Singletons" (singletons table)
filterByFrequency = filter (\x -> frequency table x >= thresh)