microsoft / CodeBERT

CodeBERT
MIT License
2.15k stars 442 forks source link

Why 'codesearch' task in CodeBERT is 'classification' #247

Open suunryo opened 1 year ago

suunryo commented 1 year ago

Follow README at CodeBERT/codesearch, I fine-tuned the model.

But, it's task type is 'classification', so, when i want to search something by model, i got a classification result, like this:

[{'label': 'LABEL_1', 'score': 0.9671146273612976}]

I mean, i want to search code, not classify my input.

guoday commented 1 year ago

you can use unixcoder, which uses retrieval setting.