How to determine the amount of layers and unit count in price prediction with Machine Learning and LSTM?

  keras, lstm, machine-learning, prediction, python

I’m a profesionnal backend developper who recently got into machine learning. I am trying to come up with an experimental solution for a problem related to prediction (similar to stock price prediction).

I’ve been reading a lot of documentation about using Machine Learning and about stock prices prediction using LSTM (keras and tensorflow). I understand the general theory of machine learning in itself and how a LSTM layer works behind the scenes.

My main concern is that every tutorial about stock price prediction I find, it comes up with a different amount of hidden LSTM layers and each one of them has a pre-defined number of units without much explanation. In the case of the classic "Apple (AAPL) stock price prediction" scenario, I see tutorials with 3 layers and 50 units, others with only 2 layers but 128 units.

In the case I have to code something by myself, how can I determine the amount of LSTM and the amount of units for the prediction to function properly? My main guess at the moment is trying to figure it out with trials and errors. I was wondering if there is some sort of relation between the problem, the data and the solution.

Also if someone can explain the effects of having more/less LSTM layers and more/less units other than doing more calculations, it would be awesome!

Thanks a lot!

Source: Python Questions

LEAVE A COMMENT