Struct rust_bert::pipelines::ner::NERModel

pub struct NERModel { /* fields omitted */ }

NERModel to extract named entities

Implementations

impl NERModel

pub fn new(
    ner_config: TokenClassificationConfig
) -> Result<NERModel, RustBertError>

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<'a, S>(&self, input: S) -> Vec<Entity> where
    S: AsRef<[&'a str]>, 

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);