soarsmu/MulaRec

The replication package of <Multi-Modal API Recommendation>. Accepted by SANER'23.

Data

Please kindly find the data link.

Training

single-annotation

CUDA_VISIBLE_DEVICES=7 python single-run.py --max_length 256 --batch_size 16 --output_dir '19-Oct-single-annotation'

single-code

CUDA_VISIBLE_DEVICES=5 python single-code-run.py --max_length 256 --batch_size 16 --output_dir '19-Oct-single-code'

Continuing traning

CUDA_VISIBLE_DEVICES=1 python single-code-run.py --max_length 256 --batch_size 16 --epoch 6 --norm True --load_model_path 18-Oct-single-code/checkpoint-best-bleu/pytorch_model.bin --output_dir '21-Oct-single-code'

dual-model

CUDA_VISIBLE_DEVICES=3 python dual-run.py --max_length 256 --batch_size 16 --epoch 30 --fuse True --norm True --output_dir '21-Oct-dual-ant-only'

continue training

CUDA_VISIBLE_DEVICES=1 python dual-run.py --max_length 256 --batch_size 16 --epoch 20 --output_dir '18-Oct-dual' --load_model_path '14-Oct-dual/checkpoint-best-bleu/pytorch_model.bin'

Evaluation

single-annotation

CUDA_VISIBLE_DEVICES=3 python evaluate.py --model_type 0 --max_length 256 --load_model_path ./14-Oct-single-256/checkpoint-best-bleu/pytorch_model.bin --test_filename ../data/test_3_lines_dedup.csv --output_dir 14-Oct-single-256/res

single-code

CUDA_VISIBLE_DEVICES=1 python evaluate.py --model_type 1 --max_length 256 --load_model_path 13-Oct-single-code/checkpoint-best-bleu/pytorch_model.bin --test_filename ../data/test_3_lines_dedup.csv --output_dir 13-Oct-single-code/res

dual-model

CUDA_VISIBLE_DEVICES=0 python evaluate.py --model_type 2 --load_model_path 14-Oct-dual/checkpoint-best-bleu/pytorch_model.bin --test_filename ../data/test_3_lines_dedup.csv --output_dir 14-Oct-dual/res-1 --max_length 256

CUDA_VISIBLE_DEVICES=0 python evaluate.py --model_type 2 --max_length 256 --load_model_path 18-Oct-dual-concat/checkpoint-best-bleu/pytorch_model.bin --test_filename ../data/test_3_lines_dedup.csv --output_dir 18-Oct-dual-concat/res

CUDA_VISIBLE_DEVICES=3 python evaluate.py --model_type 2 --load_model_path 14-Oct-dual/checkpoint-best-bleu/pytorch_model.bin --test_filename ../data/test_3_lines_dedup.csv --output_dir 14-Oct-dual/ant-only --max_length 256

CUDA_VISIBLE_DEVICES=3 python evaluate.py --model_type 2 --load_model_path 14-Oct-dual/checkpoint-best-bleu/pytorch_model.bin --test_filename ../data/test_3_lines_dedup.csv --output_dir 14-Oct-dual/code-only --max_length 256

Calculate the results

single-annotation

python calculate_bleu_score.py --reference end-to-end/13-Oct-single/res/test_ref.csv --candidate end-to-end/13-Oct-single/res/test_hyp.csv

single-code

python calculate_bleu_score.py --reference end-to-end/13-Oct-single-code/res/test_ref.csv --candidate end-to-end/13-Oct-single-code/res/test_hyp.csv

dual-model

python calculate_bleu_score.py --reference end-to-end/14-Oct-dual/res/test_ref.csv --candidate end-to-end/14-Oct-dual/res/test_hyp.csv