Source code for mowl.models.elembeddings.examples.model_gda

from mowl.models import ELEmbeddings




class ELEmGDA(ELEmbeddings):
    """
    Example of ELEmbeddings for gene-disease associations prediction.

    Uses default negative sampling (all classes for gci2) and direct loss
    with regularization inherited from ELEmbeddings.
    """

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.eval_gci_name = "gci2"



    def get_negative_sampling_config(self):
        """Only do negative sampling for gci2."""
        return {
            "gci2": {"index_pool": "classes", "corrupt_column": 2}
        }