modelTraining.scala

  var itrTrain: Int = 0
  trainDirectoryLabels.foreach(directory => {
    val label = itrTrain
    itrTrain += 1
    val images = directory.listFiles.toList
    images.foreach(image => {
      val vec = new ImageVectorizer(image, numOfTrainLabels, label)
      val imageSet: DataSet = vec.vectorize()
      
      model.fit(imageSet)
    })
  })