Please read the topic category description to understand what this is all about
Applications like GitHub’s CoPilot can automatically generate docstrings from a class or function name. The goal of this project is to fine-tune a Transformer like CodeT5 to do this ourselves!
Generating docstrings from source code can be modelled as a sequence-to-sequence task, so T5 models are a good starting point here:
A good dataset for this task is code_search_net, but feel free to find alternative datasets if you can’t find your favourite programming language there.
Models like CodeT5 are rather large and you’ll need to think about what metrics one should use for this type of task.
- Create a Streamlit or Gradio app on Spaces that can automatically generate a docstring from a class of function name in your favourite programming language!
- Don’t forget to push all your models and datasets to the Hub so others can build on them!
To chat and organise with other people interested in this project, head over to our Discord and:
Follow the instructions on the
Just make sure you comment here to indicate that you’ll be contributing to this project