dv4aby's picture
Update README
ffc3b12 verified
metadata
tags:
  - pytorch
  - code-search
  - ablation-study

concat - Procedural Code Search

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

Model Details

  • Variant: concat
  • 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.