changed indent method according to ghc standard
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
module FrequentPatterns (
|
||||
frequentPatterns
|
||||
) where
|
||||
frequentPatterns
|
||||
) where
|
||||
import DataModel
|
||||
import qualified Data.Set as Set
|
||||
import Debug.Trace (traceShow)
|
||||
@@ -8,8 +8,8 @@ import qualified Data.List as List
|
||||
|
||||
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
|
||||
max1 = Set.findMax set1
|
||||
max2 = Set.findMax set2
|
||||
max1 = Set.findMax set1
|
||||
max2 = Set.findMax set2
|
||||
|
||||
-- generate all possible combinations from a set of singletons
|
||||
-- generateLevels :: [Item] -> [[ItemSet]]
|
||||
@@ -20,14 +20,14 @@ semiUnion (ItemSet set1) (ItemSet set2) = ItemSet (if max1 <= max2 && Set.delete
|
||||
-- generate the next level in a bottom-up route
|
||||
generateNextLevel :: [ItemSet] -> [ItemSet]
|
||||
generateNextLevel level = traceShow ("Computing level " ++ show (isSize (head level))) $ foldr (\value old -> generate value ++ old) [] level where
|
||||
generate value = takeWhile (/= empty) (foldr (\x old -> semiUnion value x : old) [] (tail $ List.dropWhile (/= value) level))
|
||||
empty = ItemSet $ Set.fromList []
|
||||
isSize (ItemSet set) = Set.size set
|
||||
generate value = takeWhile (/= empty) (foldr (\x old -> semiUnion value x : old) [] (tail $ List.dropWhile (/= value) level))
|
||||
empty = ItemSet $ Set.fromList []
|
||||
isSize (ItemSet set) = Set.size set
|
||||
|
||||
singletons :: [ItemSet] -> [Item]
|
||||
singletons table = Set.toList $ foldr (\(ItemSet row) old -> old `Set.union` row) (Set.fromList []) table where
|
||||
|
||||
frequentPatterns :: Frequency -> [ItemSet] -> [[ItemSet]]
|
||||
frequentPatterns thresh table = until (\x -> [] == head x) (\x -> filterByFrequency (generateNextLevel (head x)) : x) [firstLevel] where
|
||||
firstLevel = filterByFrequency $ map (\x -> ItemSet $ Set.fromList [x]) (singletons table)
|
||||
filterByFrequency = filter (\x -> frequency table x >= thresh)
|
||||
firstLevel = filterByFrequency $ map (\x -> ItemSet $ Set.fromList [x]) (singletons table)
|
||||
filterByFrequency = filter (\x -> frequency table x >= thresh)
|
||||
|
||||
Reference in New Issue
Block a user