-
Notifications
You must be signed in to change notification settings - Fork 10
Provided jutsu.ai config options
Harley Swick edited this page Mar 28, 2018
·
2 revisions
These keywords can be placed in a config and jutsu.ai will translate them to be used with the deeplearning4j config.
(def options
{:sgd (OptimizationAlgorithm/STOCHASTIC_GRADIENT_DESCENT)
:tanh (Activation/TANH)
:identity (Activation/IDENTITY)
:mse (LossFunctions$LossFunction/MSE)
:negative-log-likelihood (LossFunctions$LossFunction/NEGATIVELOGLIKELIHOOD)
:kl-divergence (LossFunctions$LossFunction/KL_DIVERGENCE)
:relu (Activation/RELU)
:softmax (Activation/SOFTMAX)
:sigmoid (Activation/SIGMOID)
:xavier (WeightInit/XAVIER)
:rmsprop (Updater/RMSPROP)
:mcxent (LossFunctions$LossFunction/MCXENT)
:truncated-bptt (BackpropType/TruncatedBPTT)
:learning-rate-policy-schedule (LearningRatePolicy/Schedule)
:nesterovs (Updater/NESTEROVS)
:pooling-type-max (SubsamplingLayer$PoolingType/MAX)
:distribution (WeightInit/DISTRIBUTION)
:renormalize-l2-per-layer (GradientNormalization/RenormalizeL2PerLayer)
:workspace-single (WorkspaceMode/SINGLE)
:workspace-separate (WorkspaceMode/SEPARATE)
:step (LearningRatePolicy/Step)})