gluonts.nursery.spliced_binned_pareto.gaussian_model module

class gluonts.nursery.spliced_binned_pareto.gaussian_model.GaussianModel(mu, sigma, device=None)[source]

Bases: torch.nn.modules.module.Module

Model to learn a univariate Gaussian distribution.

Parameters
  • mu (Mean of the Gaussian distribution) –

  • sigma (Standard deviation of the Gaussian distribution) –

  • device (The torch.device to use, typically cpu or gpu id) –

forward(x)[source]

Takes input x as new distribution parameters

icdf(value)[source]
log_prob(x)[source]
to_device(device)[source]

Moves members to a specified torch.device

training = None