给定四个点(1,6) (2,5) (3,7) (4,10),用一条直线y=ax+b去最好的拟合这些点,令均方误差最小。
参考:
作者:努力找工作的菜鸡丁
链接:https://www.nowcoder.com/discuss/464012?type=1&channel=666&source_id=discuss_terminal_discuss_hot
来源:牛客网
vector<int> x = { 1, 2, 3, 4 };
vector<int> y = { 6, 5, 7, 10 };
int n = x.size();
double a = 0, b = 0, lr = 0.001;
while (true)
{
double g_a = 0, g_b = 0;
for (int i = 0; i < n; i++) {
g_a += 2 * x[i] * (a*x[i] + b - y[i]);
g_b += 2 * (a*x[i] + b - y[i]);
}
g_a /= n; g_b /= n;
a = a - lr * g_a;
b = b - lr * g_b;
double loss = 0;
for (int i = 0; i < n; i++) {
loss += pow((a * x[i] + b - y[i]), 2);
}
//loss卡在4.2就降不下去
if (loss < 4.2) break;
}
cout << a << endl;
cout << b << endl;
原理部分:一看就懂
https://www.cnblogs.com/paiandlu/p/7843236.html