batmangiaicuuthegioi/elph-checkpoints

---

            tags:
            - link-prediction
            - graph-neural-network
            - elph
            datasets:
            - ogbl-collab
            ---

            # ELPH Model for ogbl-collab

            This model was trained using the ELPH framework.

            ## Model Details
            - **Model**: ELPH
            - **Dataset**: ogbl-collab
            - **Run Name**: collab-raw-feat-no-embed-rank-0.002-hidden256
            - **Hidden Channels**: 256
            - **Epochs**: 20

            ## Usage

            ```python
            import torch
            from huggingface_hub import hf_hub_download

            # Download checkpoint
            checkpoint_path = hf_hub_download(
                repo_id="batmangiaicuuthegioi/elph-checkpoints",
                filename="checkpoints/ogbl-collab_ELPH_collab-raw-feat-no-embed-rank-0.002-hidden256.pt"
            )

            # Load model
            state_dict = torch.load(checkpoint_path)
            # model.load_state_dict(state_dict)
            ```