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: bool#