ESIM implementation by PyTorch, get 89.29% accuracy on Quora Questions Pairs dataset.
- Python 3.5+
- PyTorch 1.0+
- train
python run.py --gpu gpudevice(e.g. 0)
- test
python test.py ./saved_models/best.pth.tar --gpu gpudevice(e.g. 0)
The dataset, glove.npy, best.pth.tar can be downloaded from the given url in the above files.
The accuracy on the test set is 89.29%, I use the Quora Queations Pairs dataset from here, which can be downloaded from here
A keras ESIM implementation can be found here: here, which can get 88.12% accuracy on the test set.