Polynomial

make_polynomial.py

def make_polynomial(dataframe, degree=MAX_DEGREE):
    """Function for creating higher-order polynomial features from dataframe.

    Dataframe df should be like [Y, X1, X2, .. Xi].
    Returns dataframe polynomial features of X1 ... Xi up to degree polynomials."""

    df = dataframe.copy()
    cols = df.columns[1:]
    for i in range(2, degree + 1):
        for col in cols:
            df["**".join([col, str(i)])] = df[col] ** i

    return df