treeflow.model.event_shape_bijector module

treeflow.model.event_shape_bijector.inverse_with_nones(bijector: Bijector, val)

treeflow.model.event_shape_bijector.get_default_event_space_bijector(model: JointDistribution) → Composition

treeflow.model.event_shape_bijector.event_shape_fn(model: JointDistribution)

treeflow.model.event_shape_bijector.get_event_shape_and_space_bijector(model: ~tensorflow_probability.python.distributions.joint_distribution.JointDistribution, joint_bijector_func: ~typing.Callable[[~tensorflow_probability.python.distributions.joint_distribution.JointDistribution], ~tensorflow_probability.python.bijectors.composition.Composition] = <function get_default_event_space_bijector>, event_shape_fn: ~typing.Callable[[~tensorflow_probability.python.distributions.joint_distribution.JointDistribution], object] = <function event_shape_fn>) → Tuple[Bijector, Dict[str, Tensor]]

treeflow.model.event_shape_bijector.get_fixed_topology_bijector(dist: Distribution, topology_pins=typing.Dict[str, treeflow.tree.topology.tensorflow_tree_topology.TensorflowTreeTopology])

treeflow.model.event_shape_bijector.get_fixed_topology_joint_bijector(model: JointDistribution, topology_pins=typing.Dict[str, treeflow.tree.topology.tensorflow_tree_topology.TensorflowTreeTopology]) → Composition

treeflow.model.event_shape_bijector.get_fixed_topology_event_shape(model: JointDistribution, topology_pins: Dict[str, TensorflowTreeTopology])

treeflow.model.event_shape_bijector.get_fixed_topology_event_shape_and_space_bijector(model: JointDistribution, topology_pins: Dict[str, TensorflowTreeTopology])

treeflow.model.event_shape_bijector.get_unconstrained_init_values(model: ~tensorflow_probability.python.distributions.joint_distribution.JointDistribution, event_shape_and_space_bijector: ~tensorflow_probability.python.bijectors.bijector.Bijector, event_shape_fn: ~typing.Callable[[~tensorflow_probability.python.distributions.joint_distribution.JointDistribution], object] = <function event_shape_fn>, init: object | None = None)