microsoft / CodeBERT

CodeBERT
MIT License
2.19k stars 450 forks source link

How to finetune CodeBERT to do a regression prediction task #316

Open hjiang13 opened 6 months ago

hjiang13 commented 6 months ago

Hi,

Recently I have been looking and experimenting the clone detection variant of CodeBERT to perform a regression prediction task. The label of my dataset is SDC rate between 0 to 1. But it seems the model is only able to solve a classification (using codedetection folder) problem ? Any idea to finetune CodeBERT to do a regression prediction task?

Thanks a lot