Struct rust_bert::pipelines::pos_tagging::POSModel

pub struct POSModel { /* private fields */ }

POSModel to extract Part of Speech tags

Implementations

impl POSModel

pub fn new(pos_config: POSConfig) -> Result<POSModel, RustBertError>

Build a new POSModel

Arguments

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

Example

use rust_bert::pipelines::pos_tagging::POSModel;

let pos_model = POSModel::new(Default::default())?;

pub fn predict<S>(&self, input: &[S]) -> Vec<Vec<POSTag>> where
    S: AsRef<str>,

Extract entities from a text

Arguments

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

Returns

• Vec<Vec<POSTag>> containing Part of Speech tags for the inputs provided

Example

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