private val numberOfEpochs = 15
  private val printStatisticsFrequency = 50

  val model = getModel
  model.setListeners(new ScoreIterationListener(printStatisticsFrequency))

  trainModel(model)

  private def trainModel(model: MultiLayerNetwork): Unit =
    for (epoch <- 1 to numberOfEpochs) {
      println(s"Epoch --------- $epoch")
      model.fit(trainingDataSet)
    }