개발자를 위한 필수 수학

214

개발자를 위한 필수 수학

예제

5-22

선형 회귀에

3

-폴드 교차 검증 사용하기

import

pandas

as

pd

from

sklearn.linear_model

import

LinearRegression

from

sklearn.model_selection

import

KFold,

cross_val_score

df

=

pd.read_csv('https://bit.ly/3cIH97A',

delimiter=",")

# (마지막 열을 제외한 모든 열을) 입력 변수로 추출합니다.

X

=

df.values[:,

:-1]

# 마지막 열을 출력으로 추출합니다.

Y

=

df.values[:,

-1]

# 단순 선형 회귀를 수행합니다.

kfold

=

KFold(n_splits=3,

random_state=7,

shuffle=True)

model

=

LinearRegression()

results

=

cross_val_score(model,

X,

Y,

cv=kfold)

print(results)

print("MSE: 평균=%.3f (표준 편차-%.3f)"

%

(results.mean(),

results.std()))

훈련/테스트 분할은 반드시

1

/

3

이어야 하나요?

교차 검증에서

3

-폴드를 사용할 필요는 없습니다.

k

-폴드 교차 검증

k

-

fold

cross

validation

을 사용해