机器学习 -- 线性回归(Ⅳ 向量化)

该博客探讨了如何将线性回归的计算过程从使用for循环改进为向量化的运算,以提高性能效率。通过观察分子和分母的计算模式,将它们转换为向量的点乘操作,进而修改fit函数,实现了向量化的线性回归模型。最终展示的完整代码运行结果与之前的循环实现一致。
摘要由CSDN通过智能技术生成

        上节对于a的计算,使用循环的方式依次计算分子分母的m项然后进行相加。使用for循环性能效率较低,若能变成向量间的运算则性能会有较大提升。

(1)仔细观察分子分母,发现都对应于这种模式的向量相乘:

(2)转换成向量化运算:

(3)函数改进:只需要在fit函数中把循环去掉,分子分母变成对应的点乘即可。

numerator = (x_train - x_mean).dot(y_train - y_mean)
denominator = (x_train - x_mean).dot(x_train - x_mean)

(4)修改完的fit函数实现:

    # 根据训练数据集x_train,y_train训练简单线性回归模型
    def fit(self, x_train, y_train):
        assert x_train.ndim == 1, \
            "简单线性回归仅能解决单特征训练数据ÿ
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值