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) –
- training: bool#