- Data download here [12.8GB].
- Or you can download raw data from Stack Overflow data dump, the data preprocessing scripts are provided here.
| Command | |
|---|---|
| Post2Vec | |
| Train | time CUDA_VISIBLE_DEVICES=0 .env/bin/python3 src/approaches/post2vec/post2vec_train.py |
| Test | time CUDA_VISIBLE_DEVICES=0 .env/bin/python3 src/approaches/post2vec/post2vec_test.py |
| TagCNN | |
| Train | time CUDA_VISIBLE_DEVICES=0 .env/bin/python3 src/approaches/stoa/tagcnn/tagcnn_train.py |
| Test | time CUDA_VISIBLE_DEVICES=0 .env/bin/python3 src/approaches/stoa/tagcnn/tagcnn_test.py |
| TagRCNN | |
| Train | time CUDA_VISIBLE_DEVICES=5 .env/bin/python3 src/approaches/stoa/tagrcnn/tagrcnn_train.py |
| Test | time CUDA_VISIBLE_DEVICES=0 .env/bin/python3 src/approaches/stoa/tagrcnn/tagrcnn_test.py |