awslabs/gluonts

Introduce model input API

Open

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

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

描述

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.

貢獻者指南

Introduce model input API · awslabs/gluonts#2650 | Good First Issue