Struct JointModelBuilder 

pub struct JointModelBuilder { /* private fields */ }

Builder for JointModel with fluent configuration.

Example

use anno::joint::{JointModelBuilder, WikipediaKnowledgeStore};

let model = JointModelBuilder::new()
    .with_max_iterations(10)
    .with_convergence_threshold(1e-5)
    .enable_link_ner(true)
    .enable_coref_ner(true)
    .enable_coref_link(true)
    .with_knowledge(knowledge_store)
    .build()?;

Implementations

impl JointModelBuilder

pub fn new() -> Self

Create a new builder with default configuration.

pub fn with_max_iterations(self, max_iterations: usize) -> Self

Set maximum iterations for belief propagation.

pub fn with_convergence_threshold(self, threshold: f64) -> Self

Set convergence threshold for belief propagation.

pub fn with_pruning_threshold(self, threshold: f64) -> Self

Set pruning threshold for antecedent candidates.

pub fn with_max_antecedent_candidates(self, max: usize) -> Self

Set maximum antecedent candidates to keep after pruning.

pub fn with_max_link_candidates(self, max: usize) -> Self

Set maximum link candidates per mention.

pub fn enable_link_ner(self, enable: bool) -> Self

Enable or disable Link+NER factors.

pub fn enable_coref_ner(self, enable: bool) -> Self

Enable or disable Coref+NER factors.

pub fn enable_coref_link(self, enable: bool) -> Self

Enable or disable Coref+Link factors.

pub fn with_entity_types(self, types: Vec<EntityType>) -> Self

Set entity types to consider.

pub fn with_knowledge(self, store: Arc<WikipediaKnowledgeStore>) -> Self

Add Wikipedia knowledge store for semantics lookups.

pub fn with_ner_provider(self, provider: Arc<dyn NerScoreProvider>) -> Self

Plug in a custom NER score provider (unary NER factors).

pub fn with_coref_provider(self, provider: Arc<dyn CorefScoreProvider>) -> Self

Plug in a custom coreference score provider (unary coref factors).

pub fn with_link_provider(self, provider: Arc<dyn LinkScoreProvider>) -> Self

Plug in a custom link score provider (unary link factors).

pub fn build(self) -> Result<JointModel>

Build the JointModel.

Trait Implementations

impl Clone for JointModelBuilder

fn clone(&self) -> JointModelBuilder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Default for JointModelBuilder

fn default() -> JointModelBuilder

Returns the “default value” for a type.