Struct PolynomialFeatures 

pub struct PolynomialFeatures { /* private fields */ }

Polynomial Feature Generator

Generate polynomial and interaction features from input features. This transformer can be used to capture non-linear relationships by creating polynomial combinations of the original features.

Parameters

• degree - Maximum degree of polynomial features
• interaction_only - If true, only interaction features are produced
• include_bias - If true, include a bias column of ones

Examples

use sklears_feature_extraction::basic_features::PolynomialFeatures;
use sklears_core::traits::{Transform, Fit};
use scirs2_core::ndarray::array;

let X = array![[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]];
let poly = PolynomialFeatures::new()
    .degree(2)
    .include_bias(true);

let features = poly.transform(&X.view()).expect("valid input produces transform output");

Implementations

impl PolynomialFeatures

pub fn new() -> Self

Create a new PolynomialFeatures transformer

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

Set the maximum degree of polynomial features

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

Set whether to produce only interaction features

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

Set whether to include a bias column

pub fn transform(&self, X: &ArrayView2<'_, Float>) -> SklResult<Array2<Float>>

Transform input features to polynomial features

pub fn get_feature_names( &self, input_features: Option<&[String]>, ) -> SklResult<Vec<String>>

Get feature names for the polynomial features

pub fn get_n_output_features(&self, n_input_features: usize) -> SklResult<usize>

Get the number of output features

Trait Implementations

impl Clone for PolynomialFeatures

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PolynomialFeatures

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for PolynomialFeatures

fn default() -> Self

Returns the “default value” for a type.

impl Transform<ArrayBase<ViewRepr<&f64>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>> for PolynomialFeatures

fn transform(&self, X: &ArrayView2<'_, Float>) -> SklResult<Array2<Float>>

Transform the input data