mxfusion.components.distributions.bernoulli

Members

class mxfusion.components.distributions.bernoulli.Bernoulli(prob_true, rand_gen=None, dtype=None, ctx=None)

Bases: mxfusion.components.distributions.univariate.UnivariateDistribution

The Bernoulli distribution.

Parameters:
  • prob_true (Variable) – the probability of being true.
  • rand_gen (RandomGenerator) – the random generator (default: MXNetRandomGenerator).
  • dtype (numpy.float32 or numpy.float64) – the data type for float point numbers.
  • ctx (None or mxnet.cpu or mxnet.gpu) – the mxnet context (default: None/current context).
replicate_self(attribute_map=None)

This functions as a copy constructor for the object. In order to do a copy constructor we first call __new__ on the class which creates a blank object. We then initialize that object using the methods standard init procedures, and do any extra copying of attributes.

Replicates this Factor, using new inputs, outputs, and a new uuid. Used during model replication to functionally replicate a factor into a new graph.

Parameters:
  • inputs (List of tuples of name to node e.g. [('random_variable': Variable y)] or None) – new input variables of the factor.
  • outputs (List of tuples of name to node e.g. [('random_variable': Variable y)] or None) – new output variables of the factor.
log_pdf_impl(prob_true, random_variable, F=None)

Computes the logarithm of probabilistic mass function of the Bernoulli distribution.

Parameters:
  • F – MXNet computation type <mx.sym, mx.nd>.
  • prob_true (MXNet NDArray or MXNet Symbol) – the probability of being true.
  • random_variable (MXNet NDArray or MXNet Symbol) – the point to compute the logpdf for.
Returns:

log pdf of the distribution.
Rtypes:

MXNet NDArray or MXNet Symbol
draw_samples_impl(prob_true, rv_shape, num_samples=1, F=None)

Draw a number of samples from the Bernoulli distribution.

Parameters:
  • prob_true (MXNet NDArray or MXNet Symbol) – the probability being true.
  • rv_shape (tuple) – the shape of each sample.
  • num_samples (int) – the number of drawn samples (default: one).
  • F – the MXNet computation mode (mxnet.symbol or mxnet.ndarray).
Returns:

a set samples of the Bernoulli distribution
Rtypes:

MXNet NDArray or MXNet Symbol
static define_variable(prob_true, shape=None, rand_gen=None, dtype=None, ctx=None)

Creates and returns a random variable drawn from a Bernoulli distribution.

Parameters:
  • prob_true (Variable) – the probability being true.
  • shape (tuple of int) – the shape of the Bernoulli variable.
  • rand_gen (RandomGenerator) – the random generator (default: MXNetRandomGenerator).
  • dtype (numpy.float32 or numpy.float64) – the data type for float point numbers.
  • ctx (None or mxnet.cpu or mxnet.gpu) – the mxnet context (default: None/current context).
Returns:

RandomVariable drawn from the Bernoulli distribution.
Rtypes:

Variable