pyjuice.nodes.distributions.ExternProductCategorical

class pyjuice.nodes.distributions.ExternProductCategorical(num_cats: int)

A class representing a product distribution of two Categorical distributions: Pr_{A} * Pr_{B}. Pr_{A} is the the distribution parameterized by the internal parameters, similar to juice.distributions.Categorical. Pr_{B} is an external Categorical distribution provided on the fly (at training/inference time).

Parameters:

num_cats (int) – number of categories

__init__(num_cats: int)

Methods

num_param_flows()	The number of parameter flows per node.
num_parameters()	The number of parameters per node.

Attributes

need_meta_parameters

A flag indicating whether users need to pass in meta-parameters to the constructor of InputNodes.