GPT-2 medium for classifying code comments (multi-label)

This model classifies comments in Python code as 'usage', 'parameters', 'developmentNotes', 'expand' or 'summary'.

• Developed by: Fabian C. Peña, Steffen Herbold
• Finetuned from: gpt2-medium
• Replication kit: https://github.com/aieng-lab/senlp-benchmark
• Language: English
• License: MIT

Citation

@misc{pena2025benchmark,
  author    = {Fabian Peña and Steffen Herbold},
  title     = {Evaluating Large Language Models on Non-Code Software Engineering Tasks},
  year      = {2025}
}