Python3.x
pytorch>=0.4.0
numpy>=1.14.0
pandas>=0.22.0
scikit-learn>=0.14
To use this code, please do:
- Install dependencies:
pip install matplotlib numpy pandas scikit-learn
For pytorch installation, please see in PyTorch.org.
- To run the code, please run:
cd Water-Table-Depth-Prediction python3 train.py
- LSTM
- LSTM w/o dropout
- Double LSTM
- GRU
- RNN
- FFRNN
checkpoints
contains saved modelsdata
contains datasetresults
contains predictions and groundtruth water table depthplots
contains all the graphs of predictions