deep1994 / esim-pytorch

ESIM implementation by PyTorch, get 89.29% accuracy on Quora Questions Pairs dataset.

ESIM-PyTorch

ESIM implementation by PyTorch, get 89.29% accuracy on Quora Questions Pairs dataset.

Requirements

• Python 3.5+
• PyTorch 1.0+

Usage

• 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

keras ESIM

A keras ESIM implementation can be found here: here, which can get 88.12% accuracy on the test set.

Reference

• https://github.com/coetaur0/ESIM