Great! Now that we have all the preprocessing done and coded succinctly using ColumnTransformer
and Pipeline
, we can add a model. We will take the result at the end of the previous exercise, and now create a final pipeline with the ColumnTransformer
as the first step, and a LinearRegression
model as the second step.
By adding a model to the final step, the last step no longer has a .transform
method. This is the only step in a pipeline that can be a non-transformer. But now the final step also has a .predict
method, which can be called on the entire pipeline.
Instructions
Using preprocess
from the previous exercise, create a new pipeline that takes this as the first step and then a LinearRegression
model as the second.
Fit the pipeline on the training data.
Predict the pipeline on the test set.