Skip to content

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)})
Clone this wiki locally