mirror of
https://github.com/wassname/Castor.git
synced 2026-09-25 13:10:11 +08:00
Salman Mohammed
061cc8dd09
Refactored the code for Ferhan's relation prediction model using torchtext
546 B
546 B
Relation Prediction Model
- Download and extract SimpleQuestions dataset by running the script:
bash fetch_dataset.sh
- You will also require the package - torchtext.
git clone https://github.com/pytorch/text.git
cd path/to/torchtext
python setup.py install
- Run the training script with the following commands. Please check out args.py file to see the different commands available:
cd relation_prediction
python train.py
python train.py --no_cuda
python train.py --rnn_type gru