How to finetune CodeBERT to do a regression prediction task

[hjiang13]

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