Struct TextClassificationPipeline 

pub struct TextClassificationPipeline { /* private fields */ }

Pipeline for text classification

Implementations

impl TextClassificationPipeline

pub fn with_tfidf() -> Self

Create a new pipeline with a default TF-IDF vectorizer

pub fn new(vectorizer: TfidfVectorizer) -> Self

Create a new pipeline with the given vectorizer

pub fn with_feature_selector(self, selector: TextFeatureSelector) -> Self

Add a feature selector to the pipeline

pub fn fit(&mut self, dataset: &TextDataset) -> Result<&mut Self>

Fit the pipeline to training data

pub fn transform(&self, dataset: &TextDataset) -> Result<Array2<f64>>

Transform text data using the pipeline

pub fn fit_transform(&mut self, dataset: &TextDataset) -> Result<Array2<f64>>

Fit and transform in one step