Someone can explain this code in a easy way?

Viewed 29

I'm following this guided project and I stuck in this part:

step1 = ColumnTransformer(transformers=[
    ('col_tnf',OneHotEncoder(sparse=False,drop='first'),[0,1,7,10,11])
],remainder='passthrough')

step2 = LinearRegression()

pipe = Pipeline([
    ('step1',step1),
    ('step2',step2)
])

pipe.fit(X_train,y_train)

y_pred = pipe.predict(X_test)

print('R2 score',r2_score(y_test,y_pred))
print('MAE',mean_absolute_error(y_test,y_pred))

I'm confused about the meaning of the numbers and what 'col_tnf' represents.

Edit: I'm closing this question because I've already had the answer needed.

0 Answers
Related