awslabs/gluonts

Introduce model input API

Open

#2.650 aberto em 14 de fev. de 2023

Ver no GitHub
 (0 comments) (0 reactions) (0 assignees)Python (753 forks)batch import
enhancementhelp wanted

Métricas do repositório

Stars
 (3.888 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

Model networks expect a set of parameters.

For example, DeepAR (mxnet) takes the following values:

https://github.com/awslabs/gluonts/blob/9260639dd28791e48db694d409fe5ef1eeb6806d/src/gluonts/mx/model/deepar/_network.py#L874-L886

It would be useful to make these parameters available on Estimator and Predictor to expose what input values (incl. type, etc) are expected.

We already do this to some extend in torch, where we expose input_shapes and input_types on some models.

Ideally we have single way of retrieving this information from all trainable models, both mxnet and torch.

Guia do colaborador

Direção de pesquisa
Investigue os `input shapes` e `input types` existentes nos modelos torch, depois projete uma API unificada que funcione tanto para modelos mxnet quanto torch, garantindo compatibilidade reversa.
Pilha de tecnologia
python
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.
3-5 dias
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Recente
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Principalmente claro
Prerequisites
PythonGluonTS
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
30