awslabs/gluonts

Introduce model input API

Open

#2,650 建立於 2023年2月14日

在 GitHub 查看
 (0 留言) (0 反應) (0 負責人)Python (753 fork)batch import
enhancementhelp wanted

倉庫指標

Star
 (3,888 star)
PR 合併指標
 (平均合併 26天 17小時) (30 天內合併 4 個 PR)

描述

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.

貢獻者指南