gurvinder-singh-yadav / gnn-call-graphs Goto Github PK

A call graph is a control-flow graph, which represents calling relationships between subroutines in a computer program. Here we have tried to predict which function or subroutine would call other function or subroutine using Graph Neural Networks. Following GNN models are used:

• Graph Convolutional Network
• GraphSage
• Graph Isomorphism Network

• Centrality measures as node information
• Function name, parent class, parent package and structural information
• Temporal information i.e. the sequence in which function call each other

conda install mamba -c conda-forge -y
mamba init bash
mamba env create -f environment.yaml

conda activate callgraphs

• Setup the Anaconda environment
• Run the individual scripts for the 3 GCN Models