saved models are present in saved_models with corresponding base or lstm
download and place data and data_8k folder in root directory
data : https://drive.google.com/open?id=1vmyCsZolX2P4Ju3kwYY7D1uPEoZ6LXut
data_8k : https://drive.google.com/open?id=1FrwqHzM5Cv_sBImL3B7wwy4D-eACY8iR
to train:
for baseline: python train.py --epochs 40 --batch_size 512
for lstm: python train_lstm.py --epochs 40 --batch_size 128
to test:
just pass --test option without epochs and batch_size