pytorch
transformers
DGL
Code classification: CodeNet Link
Vulnerability detection: Devign link, please follow the instructions in link for preprocessing the data.
Code summarization: TLC Link
The samples with human evaluation results are stored in data/
Trained-from-scratch models: LSTM, GNN, Transformer (summarization)
Pre-trained models: CodeBERT, GraphCodeBERT, UniXCoder, PLBART (summarization)
Program classification and vulnerability detection: python run_classification_xxx.py
Code summarization: see code_sum/