Created
June 18, 2017 00:35
-
-
Save joycex99/815bcc2245f5ed67e0edbe1b524ebfda to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
(def params | |
{:test-ds-size 50000 ;; total = 284807, test-ds ~= 17.5% | |
:optimizer (adam/adam) ;; alternately, (adadelta/adadelta) | |
:batch-size 100 | |
:epoch-count 50 | |
:epoch-size 200000}) | |
(def network-description | |
[(layers/input (count (:data (first (create-dataset)))) 1 1 :id :data) ;width, height, channels, args | |
(layers/linear->relu 20) ; num-output & args | |
(layers/dropout 0.9) | |
(layers/linear->relu 10) | |
(layers/linear 2) | |
(layers/softmax :id :label)]) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment