Downloads · 30 days
0
gae-checkpoints
batmangiaicuuthegioi/gae-checkpoints
gae-checkpoints is a machine learning model from batmangiaicuuthegioi. 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 25, 2025
Repo size
243 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt243 MB · 100%
At a glance
- Access
- Public
- Created
- Dec 25, 2025
- Updated
- Dec 25, 2025
- SHA
- d96c2db3
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="batmangiaicuuthegioi/gae-checkpoints",
filename="checkpoints/ogbl-collab_GCN_collab-depth-sweep-5.pt"
)
state_dict = torch.load(checkpoint_path)
- Created
- Dec 25, 2025
- Updated
- Dec 25, 2025