더북(TheBook)
모두의 딥러닝 개정 4판
더북(TheBook)

이제 앞서 살펴본 최소 제곱법 공식 중 분모 값, 즉 ‘x의 각 원소와 x의 평균값의 차를 제곱하라’는 파이썬 명령을 만들 차례입니다. 다음과 같이 divisor라는 변수를 만들어 구현할 수 있습니다.

divisor = ➊sum([(i - mx)➋**2 ➌for i in x])

 

➊ sum( )은 ∑에 해당하는 함수입니다.

➋ 제곱을 구하라는 의미입니다.

➌ x의 각 원소를 한 번씩 i 자리에 대입하라는 의미입니다.

이제 분자에 해당하는 부분을 구하겠습니다. x와 y의 편차를 곱해서 합한 값을 구하면 됩니다. 다음과 같이 새로운 함수를 정의해서 dividend 변수에 분자 값을 저장합니다.

def top(x, mx, y, my):
    d = 0
    for i in range(len(x)):
        d += (x[i] - mx) * (y[i] - my)
    return d
dividend = top(x, mx, y, my)