|
|
--- |
|
|
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`. |
|
|
|
