Browse Source

made some stylistic changes

master
Wesley Kerfoot 11 years ago
parent
commit
bbb6c9b99a
  1. 18
      approximate.hs

18
approximate.hs

@ -10,16 +10,16 @@ import Data.Char
frequencyMap [] ngmap = ngmap frequencyMap [] ngmap = ngmap
frequencyMap ((_, 0.0):ngs) ngmap = frequencyMap ngs ngmap frequencyMap ((_, 0.0):ngs) ngmap = frequencyMap ngs ngmap
frequencyMap ((ng, p):ngs) ngmap = case Dm.lookup (DB.head ng) ngmap of frequencyMap ((ng, p):ngs) ngmap =
case Dm.lookup (DB.head ng) ngmap of
Nothing -> frequencyMap ngs (Dm.insert (DB.head ng) [(ng, p)] ngmap) Nothing -> frequencyMap ngs (Dm.insert (DB.head ng) [(ng, p)] ngmap)
Just xs -> frequencyMap ngs (Dm.insert (DB.head ng) ((ng, p) : xs) ngmap) Just xs -> frequencyMap ngs (Dm.insert (DB.head ng) ((ng, p) : xs) ngmap)
nextChoice k xs = case k of nextChoice 1 xs = DB.pack [DB.head xs]
1 -> DB.pack [DB.head xs] nextChoice _ xs = DB.take (k-1) xs
_ -> DB.take (k-1) xs
generateText k n acc cur freqmap =
generateText k n acc cur freqmap = do do
let (Just probabilities) = Dm.lookup cur freqmap let (Just probabilities) = Dm.lookup cur freqmap
choice <- MR.fromList probabilities choice <- MR.fromList probabilities
let next = DB.last choice let next = DB.last choice
@ -27,12 +27,14 @@ generateText k n acc cur freqmap = do
0 -> return acc 0 -> return acc
_ -> generateText k (n-1) (DB.append acc (nextChoice k choice)) next freqmap _ -> generateText k (n-1) (DB.append acc (nextChoice k choice)) next freqmap
gibberish n = do gibberish n =
do
js <- (DB.readFile "./quadgrams.json") js <- (DB.readFile "./quadgrams.json")
let (Just ngramPairs) = decode js :: Maybe (Dm.Map DB.ByteString Rational) let (Just ngramPairs) = decode js :: Maybe (Dm.Map DB.ByteString Rational)
result <- generateText 4 n "" 116 (frequencyMap (Dm.toList ngramPairs) (Dm.singleton 0 [])) result <- generateText 4 n "" 116 (frequencyMap (Dm.toList ngramPairs) (Dm.singleton 0 []))
return result return result
main = do main =
do
output <- gibberish 1550 output <- gibberish 1550
DB.writeFile "./output3.txt" output DB.writeFile "./output3.txt" output