Struct PolynomialFeatures 

pub struct PolynomialFeatures {
    pub degree: usize,
    pub interaction_only: bool,
    pub include_bias: bool,
    /* private fields */
}

Polynomial Feature Generator

Generates polynomial and interaction features. Example: [a, b] with degree=2 -> [1, a, b, a², ab, b²]

Fields

degree: usize

interaction_only: bool

include_bias: bool

Implementations

impl PolynomialFeatures

pub fn new(degree: usize) -> Self

pub fn interaction_only(self, value: bool) -> Self

pub fn include_bias(self, value: bool) -> Self

pub fn fit(&mut self, x: &Tensor)

Fit to determine output feature count

pub fn transform(&self, x: &Tensor) -> Tensor

Transform features to polynomial features

pub fn fit_transform(&mut self, x: &Tensor) -> Tensor