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

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.