foo.scala

  def buildNetwork(features: Int) = {
    val builder = new NeuralNetConfiguration.Builder()
      .seed(1)
      .iterations(1)
      .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
      .l1(0.1)
      .updater(Updater.NONE)
    builder.setUseRegularization(true)
    val conf = builder
      .list
      .layer(0, new OutputLayer.Builder(LossFunction.NEGATIVELOGLIKELIHOOD)
        .weightInit(WeightInit.XAVIER)
        .activation("softmax").weightInit(WeightInit.XAVIER)
        .nIn(features).nOut(2).build())
    .pretrain(false).backprop(true).build
    val network = new MultiLayerNetwork(conf)
    network.init()
    network.setListeners(Collections.singletonList[IterationListener](new ScoreIterationListener(1)))
    network
  }