通过增加 “Pipeline” 函数进行对象拟合处理。
示例代码如下:
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.linear_model import LinearRegression
# 定义pipeline对象
pipe = Pipeline([
('scaling', StandardScaler()),
('lr', LinearRegression())
])
# 训练模型
pipe.fit(X_train, Y_train)
# 在测试集上进行预测
Y_pred = pipe.predict(X_test)