File size: 638 Bytes
e7440b2 c2654c1 e7440b2 c2654c1 e7440b2 c2654c1 e7440b2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
---
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)
```
|