-
This is implemented in PyTorch with Python 2.7. The CUDA version is 8.0. You can simply run this script.
pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl
- Put the given dataset
train.txt
,dev.txt
,test.txt
in thedata
folder. - You can download Glove pretraining embedding here, and put it in the
data
folder. - Make sure some python packages are installed.
-
Simply run the following script.
python train.py
-
If you want to fine-tune your model. You can run the following script.
python train.py --resume_model=best_model_nlp.pth
-
Simply run the following script. This will create a file named
result.txt
.python test.py