Struct EntityLinker 

pub struct EntityLinker {
    pub nil_threshold: f64,
    /* private fields */
}

Links entity mentions in text to knowledge-base entities.

Fields

nil_threshold: f64

Minimum confidence threshold below which an entity is treated as NIL.

Implementations

impl EntityLinker

pub fn new() -> Self

Create an entity linker with default NIL threshold 0.1.

pub fn add_entity( &mut self, iri: impl Into<String>, label: impl Into<String>, aliases: &[&str], )

Add an entity to the knowledge base.

context is an optional textual description used for TF-IDF disambiguation.

pub fn build_index(&mut self)

Finalise the TF-IDF index (call after all entities are added).

pub fn link(&self, text: &str) -> Vec<LinkedEntity>

Find and link all entity mentions in text.

pub fn candidate_generation(&self, mention: &str) -> Vec<EntityCandidate>

Generate entity candidates matching the mention by string similarity.

pub fn disambiguate( &self, _mention: &EntityMention, candidates: &[EntityCandidate], context: &str, ) -> Option<EntityCandidate>

Disambiguate among candidates using context TF-IDF similarity.

pub fn entity_count(&self) -> usize

Number of entities in the knowledge base.

Trait Implementations

impl Default for EntityLinker

fn default() -> Self

Returns the “default value” for a type.