- original: original model
- codeBERT: using
output[0][:, 0]
- codeBERT_unfreeze: using
output[0][:, 0]
and fine-tune codeBERT
- codeBERT_unfreeze2: using
output[0][:, 0]
and fine-tune codeBERT only for the first 2 epochs
- codeBERT_unfreeze2_tandataset: codeBERT_unfreeze2 but with Tan's dataset
cd Com
python main.py -train \
-project [project name] \
-train_data [absolute data path] \
-dictionary_data [absolute data path]
cd Com
python main.py -predict \
-project [project name] \
-predict_data [absolute data path] \
-dictionary_data [absolute dictionary path] \
-load_model [absolute model path]
To train and evaluate Sim
cd Sim
python sim_model.py \
-project [project name] \
-train_data [absolute data path] \
-test_data [absolute data path]
python combination.py -project [project name]