available_supervised_models¶

def available_supervised_models() -> List[str]:
    """
    Get list of supervised models available in ProteinGym.

    Returns:
        List of model names
    """
    # 11 semi-supervised models available in ProteinGym v1.2
    models = [
        'Embeddings - Augmented - ESM1v',
        'Embeddings - Augmented - MSA Transformer',
        'Embeddings - Augmented - Tranception', 'Kermut',
        'OHE - Augmented - DeepSequence', 'OHE - Augmented - ESM1v',
        'OHE - Augmented - MSA Transformer', 'OHE - Augmented - TranceptEVE',
        'OHE - Augmented - Tranception', 'OHE - Not augmented', 'ProteinNPT'
    ]
    return models