로우코드 AI

272

로우코드 AI

8.2.3

사이킷런의 선형 회귀 모델 살펴보기

이제 노트북의 다음 셀로 이동합니다. 이 셀에는 데이터 준비,

ML

모델 학습, 모델 평가용 코

드가 모두 포함됩니다. 사이킷런의 학습 파이프라인의 개념은

7

장에서 다뤘으므로, 이 절은 모

든 코드를 자세히 설명하지 않습니다. 하지만 몇 가지 추가적인 참고 사항과 함께 간략한 개요

를 설명합니다. 먼저

import

문 다음의 데이터 처리를 담당하는 코드를 살펴보겠습니다.

y_train = train_df['sellingprice']

X_train = train_df.drop('sellingprice', axis=1)

def drop_columns(df, columns):

return df.drop(columns, axis=1)

preproc_cols = FunctionTransformer(drop_columns,

kw_args={"columns":['Unnamed: 0', 'mmr']})

numeric_columns = ['year', 'condition', 'odometer']

categorical_columns = ['make', 'model', 'trim', 'body',

'transmission', 'state', 'color', 'interior']

col_transformer = ColumnTransformer( ...