Metanyu/elph-checkpoints

---

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

            # ELPH Model for ogbl-ddi

            This model was trained using the ELPH framework.

            ## Model Details
            - **Model**: ELPH
            - **Dataset**: ogbl-ddi
            - **Run Name**: ddi-orthogonal
            - **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="Metanyu/elph-checkpoints",
                filename="checkpoints/ogbl-ddi_ELPH_ddi-orthogonal.pt"
            )

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