gluonts.mx.representation.embedding 模块

类 gluonts.mx.representation.embedding.Embedding(num_bins: int, size: Optional[int] = None, *args, **kwargs)

基类: gluonts.mx.representation.representation.Representation

表示在给定分箱之上执行嵌入操作的类。请注意，此表示旨在应用于分类/分箱数据之上。

参数

• num_bins – 应用此表示的数据的类别/分箱数量。

• size – 所需的嵌入大小。默认情况下，使用以下启发式方法：https://developers.googleblog.com/2017/11/introducing-tensorflow-feature-columns.html (默认值: round(num_bins**(1/4)))

hybrid_forward(F, data: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], observed_indicator: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], scale: Optional[Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]], rep_params: List[Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]], **kwargs) → Tuple[Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], List[Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]]]

将数据转换为所需的表示形式。

参数

• F –

• data – 目标数据。

• observed_indicator – 目标观测指标。

• scale – 预先计算的缩放因子。

• rep_params – 额外的预先计算的表示参数。

• **kwargs – 额外的块特定参数。

:param : 额外的块特定参数。

返回值

包含转换后的数据、计算出的缩放因子以及要传递给 post_transform 的额外参数的元组。

返回类型

Tuple[Tensor, Tensor, List[Tensor]]