linted files
This commit is contained in:
@@ -9,15 +9,13 @@ import Data.Map(Map)
|
||||
extractRules :: Confidence -> [ItemSet] -> [ItemSet] -> [Rule]
|
||||
extractRules threshold table patterns = filter (\x -> threshold <= confidence table x) rules where
|
||||
rules = Map.foldrWithKey (\k v old -> ruleFromSubset v k : old) [] subsets
|
||||
subsets = foldr (\x old -> insertMultiple (filteredPowerset x) x old) Map.empty patterns
|
||||
subsets = foldr (\x -> insertMultiple (filteredPowerset x) x) Map.empty patterns
|
||||
filteredPowerset (ItemSet set) = map (ItemSet . Set.fromList) $
|
||||
filter (\val -> val /= Set.toList set && val /= []) $ powerset $ Set.toList set
|
||||
|
||||
ruleFromSubset :: ItemSet -> ItemSet -> Rule
|
||||
ruleFromSubset set subset = Rule subset (difference set subset)
|
||||
|
||||
|
||||
|
||||
insertMultiple :: Ord k => [k] -> a -> Map k a -> Map k a
|
||||
insertMultiple keys value m = foldr (\x old -> Map.insert x value old) m keys
|
||||
|
||||
|
||||
Reference in New Issue
Block a user