metadata
tags:
- link-prediction
- graph-neural-network
- refined-gae
datasets:
- ogbl-collab
Refined-GAE Model for ogbl-collab
This model was trained using the Refined-GAE framework.
Model Details
- Model: GCN
- Dataset: ogbl-collab
- Run Name: collab-depth-sweep-5
- Hidden Channels: 256
- Epochs: 20
- Propagation Steps: 2
Usage
import torch
from huggingface_hub import hf_hub_download
checkpoint_path = hf_hub_download(
repo_id="AndrewNguyen/refined-gae-checkpoints",
filename="checkpoints/ogbl-collab_GCN_collab-depth-sweep-5.pt"
)
state_dict = torch.load(checkpoint_path)