Model Implementation for tse mini project 1
Encoder: ggnn_encoder.py
Decoder: ffnn_decoder.py
- Clone the repo:
git clone https://github.com/rrlcs/tse-mini-project2.git
- Extract the dataset:
gunzip data/dataset.json.gz
- Train the network:
python train.py
- Evaluate the network:
python evaluate.py
- Plot the losses:
python plot.py