Struct 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 new_with_tokenizer( pos_config: POSConfig, tokenizer: TokenizerOption, ) -> Result<POSModel, RustBertError>

Build a new POSModel with a provided tokenizer.

Arguments

• pos_config - POSConfig object containing the resource references (model, vocabulary, configuration) and device placement (CPU/GPU)
• tokenizer - TokenizerOption tokenizer to use for POS tagging.

Example

use rust_bert::pipelines::common::{ModelType, TokenizerOption};
use rust_bert::pipelines::pos_tagging::POSModel;
let tokenizer = TokenizerOption::from_file(
    ModelType::Bert,
    "path/to/vocab.txt",
    None,
    false,
    None,
    None,
)?;
let pos_model = POSModel::new_with_tokenizer(Default::default(), tokenizer)?;

pub fn get_tokenizer(&self) -> &TokenizerOption

Get a reference to the model tokenizer.

pub fn get_tokenizer_mut(&mut self) -> &mut TokenizerOption

Get a mutable reference to the model tokenizer.

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