Struct TrainingExample 

pub struct TrainingExample {
    pub text: String,
    pub mentions: Vec<JointMention>,
    pub gold_ner: HashMap<usize, EntityType>,
    pub gold_coref: HashMap<usize, Option<usize>>,
    pub gold_links: HashMap<usize, Option<String>>,
}

A training example for joint learning.

Fields

text: String

Document text

mentions: Vec<JointMention>

Mentions extracted from text

gold_ner: HashMap<usize, EntityType>

Gold NER labels (mention_idx -> EntityType)

gold_coref: HashMap<usize, Option<usize>>

Gold coreference (mention_idx -> antecedent_idx, None for new cluster)

gold_links: HashMap<usize, Option<String>>

Gold entity links (mention_idx -> KB_ID, None for NIL)

Implementations

impl TrainingExample

pub fn from_gold( text: &str, entities: &[Entity], chains: &[CorefChain], links: &[(usize, Option<String>)], ) -> TrainingExample

Create from gold annotations.

pub fn hamming_loss( &self, pred_ner: &HashMap<usize, EntityType>, pred_coref: &HashMap<usize, Option<usize>>, pred_links: &HashMap<usize, Option<String>>, ) -> f64

Compute Hamming loss between predicted and gold assignment.

Trait Implementations

impl Clone for TrainingExample

fn clone(&self) -> TrainingExample

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TrainingExample

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

Formats the value using the given formatter.