최소자승법을 이용한 학습 프로그램 - 2

이 전에 최소자승법을 이용한 예측에 대하여 글을 쓰고,

수식을 정리해 보았다.

원하는 그래프  y = a0 + a1x 에서 파라미터를 얻을 수 있는 a0, a1을 구하는 방법 또한

증명했었다.

증명한 내용을 토대로 소스코드를 작성하였다.

위에 수식에서 보면 a0와 a1을 구하기 위해 필요한 것이

xi, yi, xi*yi, xi^2에 필요함을 알 수 있다.

1. #include<stdio.h>
2.  
3. #define DATA_SIZE 2000
4.  
5. int main()
6. {
7.     unsigned char data[DATA_SIZE];
8.     double xi=0, yi=0;
9.     double sum_xi=0, sum_yi=0, sum_xiyi=0, sum_xi2=0;   //xi, yi, xiyi, xi^2
10.     double a0, a1;                      //y = a0+a1x
11.     int count=0;
12.  
13.     while(1)
14.     {
15.         if(fgets(data, DATA_SIZE, stdin)!=NULL)
16.         {
17.             if(sscanf(data,"%lf %lf", &xi, &yi)==2)
18.             {
19.                 sum_xi+=xi;
20.                 sum_yi+=yi;
21.                 sum_xiyi+=(xi*yi);
22.                 sum_xi2+=(xi*xi);
23.                 count++;
24.             }
25.         }
26.         else
27.             break;
28.     }
29.  
30.     a0 = (sum_xi2*sum_yi-sum_xi*sum_xiyi)/(count*sum_xi2-sum_xi*sum_xi);
31.     a1 = (count*sum_xiyi-sum_xi*sum_yi)/(count*sum_xi2-sum_xi*sum_xi);
32.  
33.     printf("y = %lf + %lf x", a0, a1);
34.  
35.     return 0;
36. }