haga/src/Main.hs

62 lines
1.4 KiB
Haskell
Raw Normal View History

{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE NoImplicitPrelude #-}
2019-05-30 12:40:32 +02:00
2020-01-07 08:45:50 +01:00
import Options.Applicative
import Pipes
import Pretty
2024-02-11 21:25:15 +01:00
import Protolude hiding (for)
import System.IO
-- import Szenario212Pun
2024-02-11 21:25:15 +01:00
import Szenario191
data Options = Options
{ iterations :: N,
populationSize :: N
}
2020-01-07 08:45:50 +01:00
options :: Parser Options
options =
Options
<$> option
auto
( long "iterations"
<> short 'i'
<> metavar "N"
<> value 1000
<> help "Number of iterations"
)
<*> option
auto
( long "population-size"
<> short 'p'
<> metavar "N"
<> value 100
<> help "Population size"
)
2020-01-07 08:45:50 +01:00
optionsWithHelp :: ParserInfo Options
2020-01-07 08:45:50 +01:00
optionsWithHelp =
info
(helper <*> options)
2020-01-07 08:45:50 +01:00
( fullDesc
<> progDesc "Run a GA"
<> header "haga - Haskell implementations of EAs"
)
main :: IO ()
main =
execParser optionsWithHelp >>= \opts -> do
hSetBuffering stdout NoBuffering
2024-02-11 21:25:15 +01:00
let pop = population (populationSize opts) (I prios [])
pop' <-
2024-02-11 21:25:15 +01:00
runEffect (for (run (tournament 2) 2 1 (5 / 100) pop (steps (iterations opts))) logCsv)
let (res, _) = bests 5 pop'
sequence_ $ format <$> res
where
format s = do
let f = fitness s
putErrText $ show f <> "\n" <> pretty s
2024-02-11 21:25:15 +01:00
logCsv = putText . csv
csv (t, f) = show t <> " " <> show f