ProbabilityDistribution

class maze.distributions.distribution.ProbabilityDistribution

Base class for all probability distributions.

deterministic_sample() → Any

Draw a deterministic sample from the probability distribution.

Returns:

deterministic sample tensor.

entropy() → Any

Calculate the entropy of the probability distribution.

Returns:

entropy tensor.

kl(other: ProbabilityDistribution) → Any

Calculates the Kullback-Leibler between self and the other probability distribution.

Parameters:

other – ([float]) the distribution to compare with.

Returns:

kl tensor.

log_prob(actions: Any) → Any

Returns the the log likelihood of the provided actions.

actions: the actions. :return: log likelihood tensor.

neg_log_prob(actions: Any) → Any

Returns the the negative log likelihood of the provided actions.

Parameters:

actions – the actions.

Returns:

negative log likelihood tensor.

sample() → Any

Draw a sample from the probability distribution.

Returns:

stochastic sample tensor.