gluonts.torch.distributions.quantile_output 模块

class gluonts.torch.distributions.quantile_output.QuantileOutput(quantiles: List[float])

基类: gluonts.torch.distributions.output.Output

args_dim: Dict[str, int]

domain_map(*args: torch.Tensor) → Tuple[torch.Tensor, ...]

将参数转换为正确的形状和域。

域取决于分布的类型，而正确的形状是通过重塑尾部轴获得的，以使返回的张量定义具有正确 `event_shape` 的分布。

property event_shape: Tuple

与输出对象兼容的每个独立事件的形状。

property forecast_generator: gluonts.model.forecast_generator.ForecastGenerator

in_features: int

loss(target: torch.Tensor, distr_args: Tuple[torch.Tensor, ...], loc: Optional[torch.Tensor] = None, scale: Optional[torch.Tensor] = None) → torch.Tensor

根据网络输出计算目标数据的损失。

参数

• target – 目标时间序列的值，用于计算损失。

• distr_args – 可用于构建输出分布的参数。

• loc – 分布的位置参数，可选。

• scale – 分布的尺度参数，可选。

返回

损失值，形状与 target 相同。

返回类型

损失值

property quantiles: List[float]