Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,34 @@
+---
+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-baseline
+- **Hidden Channels**: 1024
+- **Epochs**: 20
+- **Propagation Steps**: 4
+
+## 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-baseline.pt"
+)
+
+state_dict = torch.load(checkpoint_path)
+```