Struct TypeDistribution 

pub struct TypeDistribution { /* private fields */ }

Distribution over entity types.

Unlike a single EntityType prediction, this captures uncertainty by assigning probability mass to multiple types.

Invariants

• Probabilities are in [0, 1]
• Probabilities may not sum to 1 (unnormalized is allowed)
• Empty distributions are valid (no prediction)

Implementations

impl TypeDistribution

pub fn new(probs: Vec<(EntityType, f64)>) -> Self

Create a new type distribution.

Probabilities are clamped to [0, 1] but NOT normalized. This allows representing “I’m 30% sure it’s PERSON, 20% ORG, and 50% nothing”.

pub fn uniform(types: &[EntityType]) -> Self

Create a uniform distribution over types.

pub fn point_mass(entity_type: EntityType, confidence: f64) -> Self

Create a distribution with all mass on one type.

pub fn argmax(&self) -> Option<(&EntityType, f64)>

Get the most likely type and its probability.

pub fn prob(&self, entity_type: &EntityType) -> f64

Get probability for a specific type.

pub fn entropy(&self) -> f64

Get entropy of the distribution (higher = more uncertain).

Returns 0 for point mass, log(n) for uniform over n types.

pub fn margin(&self) -> f64

Get margin between top two predictions (higher = more confident).

Returns 1.0 if only one type, 0.0 if tied.

pub fn is_confident(&self, threshold: f64) -> bool

Check if the top prediction exceeds a confidence threshold.

pub fn to_map(&self) -> HashMap<EntityType, f64>

Convert to a HashMap for easier iteration.

pub fn num_types(&self) -> usize

Get the number of types with non-zero probability.

pub fn is_empty(&self) -> bool

Check if distribution is empty (no predictions).

pub fn iter(&self) -> impl Iterator<Item = (&EntityType, f64)>

Iterate over (type, probability) pairs.

Trait Implementations

impl Clone for TypeDistribution

fn clone(&self) -> TypeDistribution

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TypeDistribution

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for TypeDistribution

fn default() -> TypeDistribution

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for TypeDistribution

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for TypeDistribution

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for TypeDistribution

fn eq(&self, other: &TypeDistribution) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for TypeDistribution

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for TypeDistribution