Pedestrians.hs

import Predictor
import Scenario
import ErrorFunctions
import PedestriansLoader

import Mean
import Histogram
import FreMEn

import Cell
import CellGridMethod
import DataCollector
import PedestriansErrorEvaluator
import Trimmer

pedestriansTransform :: Method Double Float () -> Method (Cell Double) Float (CellGrid Double)
pedestriansTransform = (methodChangeType sdt id id sdh) . (cellGridMethod (\h _ -> (h, ()))) . (methodFilterDataset $ dataCollector 0 (+)) . (methodChangeType (\[t] -> t) id id id)
    where sdh (([t0,x0,y0], [ts,xs,ys]), [t,x,y]) = ((([x0,y0], [xs,ys]), [x,y]), (([t0], [ts]), [t]))
          sdt ([t, x, y], _) = ([t], [x, y])

pedestriansTransform' :: MethodWithParams p Double Float () -> MethodWithParams p (Cell Double) Float (CellGrid Double)
pedestriansTransform' = parametriseMethodTransform pedestriansTransform

spatialResolution = 0.5
temporalResolution = 300

pedestrians :: Scenario (Cell Double) Float (CellGrid Double)
pedestrians = ("Pedestrians Prediction", pedestriansLoader temporalResolution spatialResolution, pedestriansErrorEvaluator squareError, drawBarsAndArrows)

main :: IO ()
main = processScenario pedestrians
       "lincoln/trenovaci_dva_tydny.txt"
       [("lincoln/testovaci_dva_dny.txt", "Test\n2 days"), ("lincoln/testovaci_tri_dny.txt", "Test\n3 days"), ("lincoln/kontrolni_ctvrtek.txt", "Test\nThursday")]
       ( []
       >>! "Mean" :> pedestriansTransform meanMethod
       >>! "Hist" :> bruteForceTrain (pedestriansTransform' histogramMethod) (map (\a -> (86400, a)) [3,4,6,12,24]) (pedestriansErrorEvaluator squareError)
       >>! "FreMEn" :> bruteForceTrain (pedestriansTransform' $ (parametriseMethodTransform (trimmer 0 1)) freMEn) (map (\a -> (604800, 84, a)) [1..5]) (pedestriansErrorEvaluator squareError)
       )