for循环求简单线性回归的误差函数

导览全文

        了解数据集

        目标简述

        主体代码展示

01、了解数据集

data.csv：一个保存着若干数据点坐标的文件。

首先导入相关依赖，通过散点图的形式画出数据

# 导入数据库
import numpy as np
import matplotlib.pyplot as plt

# 读取文件，画出散点图
points = np.genfromtxt("data.csv",delimiter=",")
x = points[:,0]
y = points[:,1]

plt.scatter(x,y)
plt.show()

02、假设已经求出了该数据集拟合函数y = w*x + b 中的w,b的值，现在我要求封装一个函数，可以求得均方损失

03、主体代码展示

def compute_cost(w,b,points):
    total_cost = 0
    N = len(points)
    for i in range(N):
        x = points[i,0]
        y = points[i,1]
        total_cost = (y - w * x - b) ** 2
    return total_cost/N