Struct QuadraticFeatureCollection 

pub struct QuadraticFeatureCollection { /* private fields */ }

A feature collection consisting of sketched quadratic features utilizing the CountSketch-and-FFT technique described in (TODO: cite reference from paper here)

Implementations

impl QuadraticFeatureCollection

pub fn new( in_dimensions: usize, out_dimensions: usize, alpha: f32, ) -> QuadraticFeatureCollection

Constructs a new QuadraticFeatureCollection with the given number of input dimensions, the given scaling factor alpha], and the given number of quadratic features out_dimensions.

Trait Implementations

impl Clone for QuadraticFeatureCollection

fn clone(&self) -> QuadraticFeatureCollection

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl FeatureCollection for QuadraticFeatureCollection

fn get_jacobian(&self, in_vec: ArrayView1<'_, f32>) -> Array2<f32>

Given a vector in the input space of this feature space, return the Jacobian matrix for the feature vector at that point in the format f x s, for s the input space size

fn get_features(&self, in_vec: ArrayView1<'_, f32>) -> Array1<f32>

Given a vector in the input space of this feature space, return the vector of features for this feature space

fn get_in_dimensions(&self) -> usize

Return the number of input dimensions

fn get_dimension(&self) -> usize

Return the number of dimensions in the output of get_features

fn get_features_mat(&self, in_mat: ArrayView2<'_, f32>) -> Array2<f32>

Given a matrix whose rows are each input vectors, yields a new matrix where every row of the output is the featurized version of the corresponding input vector