The code used to produce the results described in the paper
To reproduce results of the paper run notebooks in the following order:
- Prepare_dataset.ipynb
- train_random_forest.ipynb
- LSTM.ipynb [optional]
- benchmark_models.ipynb
- plots.ipynb
Some hardcoded paths (dataset directory etc.) may need to be changed.
The trained LSTM models are included in the repository. To reproduce the LSTM model run LSTM.ipynb Replace the models located in ./models/lstm_models/ with your newly trained models