Prediction on new data

[surtantheta]

Can you provide a small instance of CodeT5's prediction on concode data set for the concode task ? I am not sure how to make predictions on new data and what command line argument to use.

[yuewang-cuhk]

Hi, for fine-tuning CodeT5-base on the concode task, you can go to sh folder and simply type python run_exp.py --model_tag codet5_base --task concode --sub_task none.

If you want to fine-tune on your dataset, you can add your own task in configs.py (here) and add your data path and the function to read in utils.py (here and here). The read function can be implemented in _utils.py similar to this one.