Skip to content

Instantly share code, notes, and snippets.

@Tachyon5

Tachyon5/mnnlConfig

Created February 24, 2016 18:02
Show Gist options
  • Save Tachyon5/23ba4e39e88a92acb1de to your computer and use it in GitHub Desktop.
Save Tachyon5/23ba4e39e88a92acb1de to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. Tachyon5 created this gist Feb 24, 2016.
    15 changes: 15 additions & 0 deletions mnnlConfig
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,15 @@
    MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder()
    .seed(seed)
    .gradientNormalization(GradientNormalization.ClipElementWiseAbsoluteValue)
    .gradientNormalizationThreshold(1.0)
    .iterations(iterations)
    .momentum(0.5)
    .momentumAfter(Collections.singletonMap(3, 0.9))
    .optimizationAlgo(OptimizationAlgorithm.CONJUGATE_GRADIENT)
    .list(4)
    .layer(0, new RBM.Builder().nIn(numRows*numColumns).nOut(500)
    .weightInit(WeightInit.XAVIER).lossFunction(LossFunction.RMSE_XENT)
    .visibleUnit(RBM.VisibleUnit.BINARY)
    .hiddenUnit(RBM.HiddenUnit.BINARY)
    .activation("relu")
    .build())