Safe Haskell	None
Language	Haskell2010

Run

Description

Module for running training

Synopsis

trainStep :: Tensor -> Tensor -> OpNet -> Adam -> IO (OpNet, Adam, Tensor)
trainingEpoch :: ProgressBar s -> [Tensor] -> [Tensor] -> [Tensor] -> OpNet -> Adam -> IO (OpNet, Adam, Tensor)
validStep :: Tensor -> Tensor -> OpNet -> IO Tensor
validationEpoch :: ProgressBar s -> [Tensor] -> [Tensor] -> OpNet -> [Tensor] -> IO Tensor
runEpochs :: FilePath -> Int -> Int -> Tensor -> Tensor -> Tensor -> Tensor -> OpNet -> Adam -> IO (OpNet, Adam)
run :: Args -> IO ()
runExp :: Args -> IO ()
runBase :: Args -> IO ()
testModel :: Int -> DataFrame Tensor -> [String] -> [String] -> (Tensor -> Tensor) -> IO ()
testModelExp :: DataFrame Tensor -> [String] -> [String] -> (Tensor -> Tensor) -> IO ()
region :: Int -> DataFrame Tensor -> Tensor

Documentation

trainStep :: Tensor -> Tensor -> OpNet -> Adam -> IO (OpNet, Adam, Tensor) Source #

Run one Update Step

trainingEpoch :: ProgressBar s -> [Tensor] -> [Tensor] -> [Tensor] -> OpNet -> Adam -> IO (OpNet, Adam, Tensor) Source #

Run through all Batches performing an update for each

validStep :: Tensor -> Tensor -> OpNet -> IO Tensor Source #

Run one Update Step

validationEpoch :: ProgressBar s -> [Tensor] -> [Tensor] -> OpNet -> [Tensor] -> IO Tensor Source #

Run through all Batches performing an update for each

runEpochs :: FilePath -> Int -> Int -> Tensor -> Tensor -> Tensor -> Tensor -> OpNet -> Adam -> IO (OpNet, Adam) Source #

Run Training and Validation for a given number of Epochs

Initiate Training Run for given Args

Experimental Training Run

Basic Training Run

testModel :: Int -> DataFrame Tensor -> [String] -> [String] -> (Tensor -> Tensor) -> IO () Source #

Model Test

testModelExp :: DataFrame Tensor -> [String] -> [String] -> (Tensor -> Tensor) -> IO () Source #

Experimental Model Test

region :: Int -> DataFrame Tensor -> Tensor Source #

Filter region Experimental: -1 Saturation: 2 Sub-Threshold: 3