awslabs/gluonts

Introduce model input API

Open

#2650 aperta il 14 feb 2023

Vedi su GitHub
 (0 commenti) (0 reazioni) (0 assegnatari)Python (753 fork)batch import
enhancementhelp wanted

Metriche repository

Star
 (3888 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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.

Guida contributor

Direzione di ricerca
Esamina gli `input shapes` e `input types` esistenti nei modelli torch, quindi progetta un'API unificata che funzioni sia per i modelli mxnet che torch, assicurando la compatibilità all'indietro.
Tech stack
python
Dominio
machine learning
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
3-5 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Nuova
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Abbastanza chiara
Prerequisiti
PythonGluonTS
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
30