Downloads · 30 days
0
refined-gae-checkpoints
AndrewNguyen/refined-gae-checkpoints
refined-gae-checkpoints is a machine learning model from AndrewNguyen. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model was trained using the Refined-GAE framework.
Access
Public
Updated Dec 26, 2025
Repo size
508 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt265 MB · 100%
At a glance
- Access
- Public
- Created
- Dec 26, 2025
- Updated
- Dec 26, 2025
- SHA
- 53c5ac96
Datasets
Model card
From the Hugging Face model README
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)
- Created
- Dec 26, 2025
- Updated
- Dec 26, 2025