Use the python code generation example in the paper

salesforce / CodeT5

Home of CodeT5: Open Code LLMs for Code Understanding and Generation

https://arxiv.org/abs/2305.07922

BSD 3-Clause "New" or "Revised" License

2.68k stars 394 forks source link

Use the python code generation example in the paper #36

Closed zhengzhongliang closed 2 years ago

zhengzhongliang commented 2 years ago

Hi, I was trying to find the model to fulfill the example in the paper "Generate Python: increment value" -> "def inc_value(x):...".

May I know if there is a released model to do this (no need to fine-tune)? In the readme file the only text-to-code generation model seems to be the concode model, which generates Java code.

If there is not, may I know what is the recommended model to do the fine-tuning to fulfill this task?

Thanks!

yuewang-cuhk commented 2 years ago

Hi, currently there is no released model to do this. You can directly fine-tune the released CodeT5-small/base model on the python subset of CodeSearchNet data for a text-to-code generation task.