PolynomialFeatures

Генерирует полиномиальные признаки и их взаимодействия.

СигнатураPolynomialFeatures(degree=2, include_bias=True, interaction_only=False)

Создаёт новые признаки как степени и попарные произведения исходных, что позволяет линейным моделям улавливать нелинейные зависимости. Степень задаётся параметром degree.

from sklearn.preprocessing import PolynomialFeatures

poly = PolynomialFeatures(degree=2, include_bias=False)
X = [[2, 3]]
print(poly.fit_transform(X))  # [2, 3, 4, 6, 9]

← Все записи: scikit-learn