awslabs/gluonts

Introduce model input API

オープン

#2,650 opened on 2023/02/14

 (0 件のコメント) (0 件のリアクション) (0 人の担当者)Python (753 件のフォーク)batch import
enhancementhelp wanted

Repository metrics

Stars
 (3,888 個のスター)
PR merge metrics
 (平均マージ 26d 17h) (30d で 4 merged PRs)

説明

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.

コントリビューターガイド