pub trait FeatureGenerator: Send + Sync {
// Required methods
fn generate(&self, data: &Array2<f64>) -> Result<Array2<f64>, SklearsError>;
fn output_dim(&self) -> usize;
fn name(&self) -> &str;
// Provided methods
fn is_stateful(&self) -> bool { ... }
fn fit_generator(&mut self, _data: &Array2<f64>) -> Result<(), SklearsError> { ... }
}Expand description
Feature generator trait
Required Methods§
Sourcefn generate(&self, data: &Array2<f64>) -> Result<Array2<f64>, SklearsError>
fn generate(&self, data: &Array2<f64>) -> Result<Array2<f64>, SklearsError>
Generate features from input data
Sourcefn output_dim(&self) -> usize
fn output_dim(&self) -> usize
Get the output dimension
Provided Methods§
Sourcefn is_stateful(&self) -> bool
fn is_stateful(&self) -> bool
Check if generator is stateful (needs fitting)
Sourcefn fit_generator(&mut self, _data: &Array2<f64>) -> Result<(), SklearsError>
fn fit_generator(&mut self, _data: &Array2<f64>) -> Result<(), SklearsError>
Fit the generator if stateful