Struct ConfusableSet 

pub struct ConfusableSet {
    pub surface_form: String,
    pub entities: Vec<ConfusableEntity>,
}

A set of confusable entities for a given surface form.

Fields

surface_form: String

The surface form (e.g., “John Smith”)

entities: Vec<ConfusableEntity>

All confusable entities

Implementations

impl ConfusableSet

pub fn new(surface_form: &str) -> ConfusableSet

Create a new confusable set.

pub fn add( &mut self, kb_id: &str, reason: ConfusableReason, description: Option<&str>, )

Add a confusable entity.

pub fn add_entity(&mut self, entity: ConfusableEntity)

Add a full confusable entity.

pub fn to_training_pairs(&self, positive_kb_id: &str) -> Vec<TrainingPair>

Generate training pairs for contrastive learning.

Given a positive entity ID, returns (positive, negative) pairs for all other confusable entities.

pub fn len(&self) -> usize

Number of confusable entities.

pub fn is_empty(&self) -> bool

Check if empty.

pub fn filter_by_reason( &self, reason: &ConfusableReason, ) -> Vec<&ConfusableEntity>

Get entities by reason.

Trait Implementations

impl Clone for ConfusableSet

fn clone(&self) -> ConfusableSet

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ConfusableSet

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

Formats the value using the given formatter.

impl Default for ConfusableSet

fn default() -> ConfusableSet

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for ConfusableSet

fn deserialize<__D>( __deserializer: __D, ) -> Result<ConfusableSet, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for ConfusableSet

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

where __S: Serializer,

Serialize this value into the given Serde serializer.