fix(notmnist): notmnist sigmoid chart with 10 epochs

This commit is contained in:
Mahdi Dibaiee
2016-09-13 10:04:30 +04:30
parent 6b9cb1fa3e
commit c0083f5c05
5 changed files with 11 additions and 4 deletions

View File

@@ -84,6 +84,7 @@ module Sibe
, batchSize :: Int
, chart :: [(Int, Double, Double)]
, momentum :: Double
, biases :: Bool
} deriving (Show)
emptyNetwork = randomNetwork 0 (0, 0) 0 [] (0, (id, id))
@@ -97,6 +98,7 @@ module Sibe
, batchSize = 0
, chart = []
, momentum = 0
, biases = True
}
saveNetwork :: Network -> String -> IO ()
@@ -122,6 +124,9 @@ module Sibe
runLayer :: Input -> Layer -> Output
runLayer input (Layer !biases !weights _) = input <# weights + biases
runLayer' :: Input -> Layer -> Output
runLayer' input (Layer !biases !weights _) = input <# weights
forward :: Input -> Session -> Output
forward input session = compute input (network session)
where