Stefano d'Apolito, Francesco Forcher, Simeone de Fremond, Alper Karaaslan
In this project we have explored the use of imaging time series to enhance forecasting results with Neural Networks. The approach has revealed itself to be extremely promising as, both in combination with an LSTM architecture and without, it has out-performed the pure LSTM architecture by a solid margin within our test datasets.
- Python 3.6
- Tensorflow 2
- Tensorboard
- Keras
- H5py
- Matplotlib
- Numba
- Numpy
- Pandas
- Pyts
- Scipy & Sklearn
- Tqdm
To install all requirements run:
pip install -r requirements.txt
If you run into some issues, make sure that matplotlib is already installed on your machine.
Our models have been implemented for two generated datasets and an additional real life dataset, for which trained models are provided. The logs, predictions, and model weights can all be found in their appropriate folders in the final_results directory.
To run the training and testing of the model:
python script\example_time_series.py --model=model (--hw) (--no_cb)
where model can be:
- "Conv_LSTM"
- "Conv_MLP"
- "LSTM" (in this case, the value of hw does not matter)
if --hw is supplied: the chosen model will be combined with Holt-Winter otherwise it will be just pure NN if --no_cb is supplied: no callbacks will be recorded, only a csv file containing predictions will be recorded at the end
Example: python script\example_time_series.py --model="Conv_LSTM" --hw
(Warning: because of the number of datapoints we train, it may take up to 25Gb of RAM)
To only get the results of our experiments run:
python script\example_time_series_results.py
To run the training and testing of the model:
python script\example_lorenz.py --model=model (--hw) (--no_cb)
where model can be:
- "Conv_LSTM"
- "Conv_MLP"
- "LSTM" (in this case, the value of hw does not matter)
if --hw is supplied: the chosen model will be combined with Holt-Winter otherwise it will be just pure NN if --no_cb is supplied: no callbacks will be recorded, only a csv file containing predictions will be recorded at the end
Example: python script\example_lorenz.py --model="Conv_LSTM" --hw
(Warning: because of the number of datapoints we train, it may take up to 25Gb of RAM)
To only get the results of our experiments run:
python script\example_lorenz_results.py
To run the training and testing of the model:
python script\Dowjones_Executer.py --model=model (--hw) (--no_cb)
where model can be:
- "Conv_LSTM"
- "Conv_MLP"
- "LSTM" (in this case, the value of hw does not matter)
if --hw is supplied: the chosen model will be combined with Holt-Winter otherwise it will be just pure NN if --no_cb is supplied: no callbacks will be recorded, only a csv file containing predictions will be recorded at the end
Example: python script\Dowjones_Executer.py --model="Conv_LSTM" --hw
To only get the results of our experiments run:
python script\dowjones_results.py
Below are some sample results of our generated predictions vs the ground truth.
This project was awarded 5.75/6.0 as part of the Deep Learning class taught by Prof. Dr. Thomas Hofmann at ETH Zürich.