[−][src]Struct rust_bert::pipelines::sentiment::SentimentModel
Methods
impl SentimentModel
[src]
pub fn new(sentiment_config: SentimentConfig) -> Fallible<SentimentModel>
[src]
Build a new SentimentModel
Arguments
sentiment_config
-SentimentConfig
object containing the resource references (model, vocabulary, configuration) and device placement (CPU/GPU)
Example
use rust_bert::pipelines::sentiment::SentimentModel; let sentiment_model = SentimentModel::new(Default::default())?;
pub fn predict(&self, input: &[&str]) -> Vec<Sentiment>
[src]
Extract sentiment form an array of text inputs
Arguments
input
-&[&str]
Array of texts to extract the sentiment from.
Returns
Vec<Sentiment>
Sentiments extracted from texts.
Example
use rust_bert::pipelines::sentiment::SentimentModel; let sentiment_classifier = SentimentModel::new(Default::default())?; let input = [ "Probably my all-time favorite movie, a story of selflessness, sacrifice and dedication to a noble cause, but it's not preachy or boring.", "This film tried to be too many things all at once: stinging political satire, Hollywood blockbuster, sappy romantic comedy, family values promo...", "If you like original gut wrenching laughter you will like this movie. If you are young or old then you will love this movie, hell even my mom liked it.", ]; let output = sentiment_classifier.predict(&input);
Auto Trait Implementations
impl !RefUnwindSafe for SentimentModel
impl !Send for SentimentModel
impl !Sync for SentimentModel
impl Unpin for SentimentModel
impl !UnwindSafe for SentimentModel
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized,
[src]
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T
[src]
impl<T> From<T> for T
[src]
impl<T, U> Into<U> for T where
U: From<T>,
[src]
U: From<T>,
impl<T, U> TryFrom<U> for T where
U: Into<T>,
[src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>
[src]
impl<T, U> TryInto<U> for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,