Pipline的使用
2018-11-04 本文已影响0人
61etj
Pipline
当我们有很多数据需要做一段类似的操作的时候我们就可以使用Pipline管道
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler
poly_reg = Pipeline([
("poly", PolynomialFeatures(degree=2)),
("std_scaler", StandardScaler()),
("lin_reg", LinearRegression())
])
poly_reg.fit(X, y)
y_predict = poly_reg.predict(X)
程序会顺序执行先进行多项式回归增加维度,然后再进行归一化,最后进行线性回归