Использование конвейеров Scikit-Learn для объединения трансформатора и оценщика - PullRequest
0 голосов
/ 19 сентября 2018

Я пытаюсь использовать функцию конвейера Scikit-Learn для организации наших преобразователей и оценщиков, и у меня возникают проблемы с построением конвейера, который комбинирует one_hot_transformer с оценщиком LinearRegression ().Сложно подключить следующие

from sklearn.preprocessing import OneHotEncoder
cat_feats = np.array([[1,10],[2,20],[3,10],[4,20],[3,10],[2,20],[1,10]])
OneHotEncoder(sparse=False).fit_transform(cat_feats)
one_hot_transformer = OneHotEncoder(sparse=False).fit_transform(X,y)

from sklearn.pipeline import Pipeline
linear_est = Pipeline([one_hot_transformer], LinearRegression())
linear_est.fit(X,y)
predicted = linear_est.predict(X)
grader.score('intro_ml__linear_model', linear_est.predict)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...