Struct rust_bert::pipelines::ner::NERModel

pub struct NERModel { /* fields omitted */ }

NERModel to extract named entities

Methods

impl NERModel

pub fn new(ner_config: NERConfig) -> Fallible<NERModel>

Build a new NERModel

Arguments

• ner_config - NERConfig object containing the resource references (model, vocabulary, configuration) and device placement (CPU/GPU)

Example

use rust_bert::pipelines::ner::NERModel;

let ner_model =  NERModel::new(Default::default())?;

pub fn predict(&self, input: &[&str]) -> Vec<Entity>

Extract entities from a text

Arguments

• input - &[&str] Array of texts to extract entities from.

Returns

• Vec<Entity> containing extracted entities

Example

let ner_model =  NERModel::new(Default::default())?;
let input = [
    "My name is Amy. I live in Paris.",
    "Paris is a city in France."
];
let output = ner_model.predict(&input);