6 파이썬 코딩으로 확인하는 평균 제곱 오차

이제 앞서 알아본 평균 제곱 오차를 파이썬으로 구현해 보겠습니다.

임의로 정한 기울기 a와 y 절편 b의 값이 각각 3과 76이라고 할 때, 가상의 기울기가 fake_a, 가상의 y 절편이 fake_b인 함수식 predict()를 다음과 같이 정의할 수 있습니다.

fake_a = 3
fake_b = 76

def predict(x):
    return fake_a * x + fake_b

위 코드의 결괏값이 들어갈 빈 리스트를 만듭니다.

predict_result = []

이제 모든 x 값을 predict() 함수에 한 번씩 대입해 예측 값 리스트를 채우는 코드를 다음과 같이 작성합니다.

for i in range(len(x)):
    predict_result.append(predict(x[i]))
    print("공부시간=%.f, 실제점수=%.f, 예측점수=%.f" % (x[i], y[i], predict(x[i])))