tags:
- link-prediction
- graph-neural-network
- elph
datasets:
- ogbl-collab
---
# ELPH Model for ogbl-collab
This model was trained using the ELPH framework.
## Model Details
- **Model**: ELPH
- **Dataset**: ogbl-collab
- **Run Name**: collab-raw-feat-no-embed-rank-0.002-hidden256
- **Hidden Channels**: 256
- **Epochs**: 20
## Usage
```python
import torch
from huggingface_hub import hf_hub_download
# Download checkpoint
checkpoint_path = hf_hub_download(
repo_id="batmangiaicuuthegioi/elph-checkpoints",
filename="checkpoints/ogbl-collab_ELPH_collab-raw-feat-no-embed-rank-0.002-hidden256.pt"
)
# Load model
state_dict = torch.load(checkpoint_path)
# model.load_state_dict(state_dict)
```