main class

main class

object SparkLSTM {
  def main(args: Array[String])= {
    if (args.length == 2) {
      val filePath = args(0)    //"/Users/kym1992/STUDY/NEU/CSYE7200/Dataset/nyse/prices-split-adjusted.csv"
      val symbolName = args(1)
      BasicConfigurator.configure()
      val prepared = StockPricePredictionLSTM.prepare(filePath, symbolName, 0.90)
      val result = StockPricePredictionLSTM.predictPriceOneAhead(prepared._1, prepared._2, prepared._3, prepared._4, prepared._5)
      println("predicts, actual")
      (result.predicts, result.actuals).zipped.foreach((x, y) => println(x + ", " + y))
      saveAsCsv(result, symbolName)

      result.predicts.foreach(r => println(r))
    }
  }