Skip to content

ErykDev/StockPricePredictionDL4J

Repository files navigation

StockPricesPredictionDL4J

Lstm based stock price prediction

Example result

How to run

  • Import project dependencies with Gradle
  • If you own an NVIDIA card with a CUDA support or have cpu that supports avx2 you can enable the usage in Gradle file

Model trained on NSE-TATAGLOBAL is avilable here

Training

  • The project is using a csv data-format. See example
  • If you wish to use your own data simply paste it into the project folder in csv format if you have a custom csv edit StockCSVDataSetFetcher.java

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.