机器学习-线性回归1

记得晒枕头

于 2020-04-04 10:14:43 发布

阅读量128

点赞数

分类专栏：机器学习文章标签：机器学习

本文链接：https://blog.csdn.net/weixin_44846884/article/details/105306415

版权

机器学习专栏收录该内容

0 篇文章 0 订阅

订阅专栏

线性回归

前言

个人学习笔记，仅供本人复习使用
主要内容来自liuyubobobo
第一次记录时间：2020年4月4日
###简单实现

import numpy as np
import matplotlib.pyplot as plt
x = np.array([1., 2., 3., 4., 5.])
y = np.array([1., 3., 2., 3., 5.])
plt.scatter(x, y)
plt.axis([0, 6, 0, 6])

在这里插入图片描述

x_mean = np.mean(x)
y_mean = np.mean(y)
num = 0.0 #分子
d = 0.0 #分母

for x_i, y_i in zip(x ,y):
  num += (x_i - x_mean) * (y_i - y_mean)
  d += (x_i - x_mean) ** 2
  a = num / d
b = y_mean - a * x_mean
y_hat = a * x + b
plt.scatter(x, y)
plt.plot(x, y_hat, color="r")
plt.axis([0, 6, 0, 6])