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

```python
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)
```