dv4aby/Procedural-Code-Search-OnlyGraph-POJ104

only_graph - Procedural Code Search

This is the only_graph variant from the ablation study for the Structural Encoder V2.

Model Details

• —Variant: only_graph
• —Dataset: POJ-104
• —Architecture:
• —only_graph: Pure Structural Encoder (R-GNN only)
• —concat: GraphCodeBERT + R-GNN (Concatenation Fusion)
• —v2: GraphCodeBERT + R-GNN (Gated Fusion)

Usage

This model requires the custom code structural_encoder_ablation.py and structural_encoder_v2.py.