Struct RFE 

pub struct RFE {
    pub n_features_to_select: usize,
    pub step: usize,
    /* private fields */
}

Recursive Feature Elimination

Fields

n_features_to_select: usize

step: usize

Implementations

impl RFE

pub fn new(n_features_to_select: usize) -> Self

pub fn step(self, step: usize) -> Self

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

Fit RFE using feature importance from a simple model

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

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

pub fn get_support(&self) -> Vec<bool>

pub fn ranking(&self) -> Vec<usize>