Update README

0f0d8c7 verified about 1 month ago

728 Bytes

metadata

tags:
  - pytorch
  - code-search
  - ablation-study

reranker - Procedural Code Search

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

Model Details

Variant: reranker
Dataset: POJ-104
Architecture:
- only_graph: Pure Structural Encoder (R-GNN only)
- concat: GraphCodeBERT + R-GNN (Concatenation Fusion)
- only_graph: Pure Structural Encoder (R-GNN only)
- concat: GraphCodeBERT + R-GNN (Concatenation Fusion)
- v2: GraphCodeBERT + R-GNN (Gated Fusion)
- reranker: Stage B Procedural Similarity Reranker (Logistic Regression)

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