JaxCategoricalCredalSet

class probly.representation.credal_set.jax.JaxCategoricalCredalSet(*args, **kwargs)[source]

Bases: CategoricalCredalSet, ABC

Base class for jax-backed categorical credal sets.

abstract property barycenter: T

Return the barycenter of the credal set.

abstractmethod classmethod from_jax_sample(sample: JaxSample[JaxCategoricalDistribution]) → Self[source]

Create a credal set from categorical distribution samples.

classmethod from_sample(sample: Sample[JaxCategoricalDistribution]) → Self[source]

Create a credal set from a finite sample.

Parameters:

sample – The sample to create the credal set from.

Returns:

The created credal set.

abstract property num_classes: int

Return the number of classes in the credal set.

type: CredalSetType = 'categorical'