PolynomialFeatureGenerator

class mastml.feature_generators.PolynomialFeatureGenerator(features=None, degree=2, interaction_only=False, include_bias=True)[source]

Bases: BaseGenerator

Class to generate polynomial features using scikit-learn’s polynomial features method More info at: http://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.PolynomialFeatures.html

Args:

degree: (int), degree of polynomial features

interaction_only: (bool), If true, only interaction features are produced: features that are products of at most degree distinct input features (so not x[1] ** 2, x[0] * x[2] ** 3, etc.).

include_bias: (bool),If True (default), then include a bias column, the feature in which all polynomial powers are zero (i.e. a column of ones - acts as an intercept term in a linear model).

Methods:
fit: conducts fit method of polynomial feature generation
Args:

df: (dataframe), dataframe of input X and y data

transform: generates dataframe containing polynomial features
Args:

df: (dataframe), dataframe of input X and y data

Returns:

(dataframe), dataframe containing new polynomial features, plus original features present

Methods Summary

fit(X[, y])

transform(X)

Methods Documentation

fit(X, y=None)[source]
transform(X)[source]