как удалить столбцы только для взаимодействия из sklearn.preprocessing.PolynomialFeatures

Согласно приведенному ниже описанию: «если входная выборка является двумерной и имеет форму [a, b], полиномиальные признаки степени 2 равны [1, a, b, a^2, ab, b^2]. '

sklearn.preprocessing.PolynomialFeatures< /а>

Мне нужно, чтобы сгенерированный выходной ndarray имел тип: [a, b, a^2, b^2]. Я знаю, что poly = PolynomialFeatures(2, include_bias=False) poly.fit_transform(X) дает вывод, например: [a, b, a^2, ab, b^2]. Но мне не нужны эти промежуточные столбцы типа «ab», которые генерируются. Как это сделать? Или любой лучший API, который можно использовать здесь?


scikit-learn preprocessor polynomials non-linear-regression
person user3868051    schedule 10.02.2018    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Ознакомьтесь с patsy и этой интеграцией scikit-learn patsylearn.

Это дает вам полный контроль над R-подобной формулой:

from patsylearn import PatsyTransformer
transformer = PatsyTransformer("y ~ a + b + a^2 + b^2")
transformer.fit(data)
person Marcus V.    schedule 11.02.2018