pyg-team/pytorch_geometric

Graph Positional and Structural Encoder

Open

#8.310 aberto em 1 de nov. de 2023

Ver no GitHub
 (21 comments) (0 reactions) (0 assignees)Python (3.514 forks)batch import
featuregood first issuehelp wantednn

Métricas do repositório

Stars
 (19.985 stars)
Métricas de merge de PR
 (Mesclagem média 16d 3h) (13 fundiu PRs em 30d)

Description

🚀 The feature, motivation and pitch

It would be great to add the "GPSE" (Graph Positional and Structural Encoder) module to PyG, which acts as a trained encoder for Graph Transformers. Most importantly, we should support loading pre-trained encoders and provide them to the user, e.g., via

encoder = GPSE.from_pretrained("molpcba")

Code: https://github.com/G-Taxonomy-Workgroup/GPSE Paper: https://arxiv.org/abs/2307.07107 Twitter: https://twitter.com/RemyLau3/status/1717194030986190899

Any help on this is highly appreciated :)

Alternatives

No response

Additional context

No response

Guia do colaborador

Direção de pesquisa
Estude o artigo GPSE e a implementação de referência, depois integre o mecanismo de carregamento do codificador pré treinado no zoo de modelos do PyG.
Pilha de tecnologia
pythonpytorch
Domain
machine learningapi
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Meio dia
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorchGit
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
50