实验楼机器学习挑战赛-------最小二乘法 Python 实现

介绍

目标

根据上面的最小二乘法代数推导过程，使用 Python 实现样本 x, y 的最小二乘法求解的函数。

def least_squares_function(x, y):
    """
    参数:
    x -- 样本 x 取值列表
    y -- 样本 y 取值列表

    返回:
    w0 -- 线性方程参数, 保留两位小数
    w1 -- 线性方程参数, 保留两位小数
    """

    ### TODO ###

    return w0, w1 # 务必按此顺序返回

要求

• 题目需使用 Python3 完成，不能使用标准库及第三方库。
• 传入参数为列表类型，例如：x=[1,2,3,4]; y=[4,5,6,7]。
• 返回参数为四舍五入保留 2 位浮点数。

代码

def least_squares_function(x, y):
    """
    参数:
    x -- 样本 x 取值列表
    y -- 样本 y 取值列表

    返回:
    w0 -- 线性方程参数, 保留两位小数
    w1 -- 线性方程参数, 保留两位小数
    """

    ### TODO ###
    sumxy=0
    for i in x:
        for j in y:
            sumxy=sumxy+x*y

    sumx=0
    sumy=0
    for i in x:
        sumx=sumx+i

    for i in y:
        sumy=sumy+i
    sumx2=0
    for i in x:
        sumx2=sumx2+i**2

    w1=(len(x)*sumxy-(sumx*sumy))/(len(x)*sumx2-sumx**2)
    w0=((sumx2*sumy)-(sumx*sumxy))/(len(x)*sumx2-sumx**2)
    w1=round(w1,2)
    w0=round(w0,2)



    return w0, w1 # 务必按此顺序返回