Skip to content

Instantly share code, notes, and snippets.

@joycex99
Created June 18, 2017 00:35
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save joycex99/815bcc2245f5ed67e0edbe1b524ebfda to your computer and use it in GitHub Desktop.
Save joycex99/815bcc2245f5ed67e0edbe1b524ebfda to your computer and use it in GitHub Desktop.
(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