pyg-team/pytorch_geometric

Weighted aggregations `torch_geometric.nn.aggr.Aggregation`

Open

#6.867 aberto em 7 de mar. de 2023

Ver no GitHub
 (10 comments) (0 reactions) (0 assignees)Python (3.514 forks)batch import
1 - Priority P1featurehelp wantednn

Métricas do repositório

Stars
 (19.985 stars)
Métricas de merge de PR
 (Mesclagem média 35d 1h) (14 fundiu PRs em 30d)

Description

🚀 The feature, motivation and pitch

The current API of torch_geometric.nn.aggr.Aggregation seems to be unnecessarily limited with respect to edge weights. At least some architectures introduce edge weights, with GCN being the most prominent one. For mean aggregation, it is simple to work around this with a custom message passing / convolution that multiplies features with edge weights. However, for most other weighted aggregations (e.g. median, quantiles, standard deviation, ...) it is not.

Alternatives

Add edge weights to aggregations API

Additional context

No response

Guia do colaborador

Direção de pesquisa
Investigue a implementação atual das classes Aggregation em torch geometric.nn.aggr. Identifique onde os pesos das arestas não são suportados. Examine as soluções alternativas existentes no repositório (por exemplo, convolução GCN). Proponha um design para adicionar suporte a pesos das arestas em agregações como median e quantiles. Discuta no tópico da issue.
Pilha de tecnologia
pythonpytorch
Domain
machine learning
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.
1-2 dias
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
PythonPyTorchGNN basics
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
65