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