This project provides a collection of datasets and source code which are used in our DeGraphCS model. The content of project is as follows:
Dataset
DeGraphCS Source Code
Variable-based Flow Graph Construction
Baseline methods
User Study
Appendix
To help people to reproduce our work, we provide raw datasets which are consist of C code snippet, corresponding code comment and generated IR.
The raw datasets can be accessed in Google Drive
To feed into our model, we first generate Variable-based Flow Graph of 41152 methods and extract corresponding comments. Then we split the datasets into 39152 training set and 2000 test set. All of the data are puted in dataset/ directory.
We provide DeGraphCS model code which are listed in src/ directory.
To construct Variable-based Flow Graph according to llvm IR, We provide graph construction code to help users to generate graph which are puted in IR2graph/ directory.
We have reproduced other code search works which are putted in Baseline methods/ directory.
We make a user study to evaluate our model.
50 queries of the user study are listed in the user study/queries.txt. And according to four models (UNIF, MMAN, DeepCS and DeGraphCS), we obtain corresponding searching result which are listed in user study/ directory.
Run the command to split comments datasets into training set and test set, and build dictionary
python src/util_desc.py
Run the command to split Variable-based Flow Graph datasets into training set and test set, and build dictionary
python src/util_ir.py
python src/train.py
python src/test.py