Python 线性回归模型

最新推荐文章于 2024-09-08 11:03:33 发布

(((φ(◎ロ◎;)φ)))牵丝戏安

最新推荐文章于 2024-09-08 11:03:33 发布

阅读量1.7k

点赞数

文章标签： python 机器学习深度学习

本文链接：https://blog.csdn.net/weixin_51341142/article/details/123420261

版权

代码段


#coding=UTF-8
from   numpy import *
import  numpy as np
import matplotlib.pyplot as plt

x=np.array([[1,5.56],[2,5.70],[3,5.91][4,6.40],[5,6.80]
               ,[6,7.05],[7,8.90],[8,8.70],[9,9.00],[10,9.05]])
m,n=np.shape(x)
x_data=np.ones((m,n))
x_data[:,:-1]=x[:,:-1]
y_data=x[:,-1]
m,n=np.shape(x_data)
theta=np.ones(n)

def gradientDescent(iter,x,y,w,alpha):
               x_train=x.transpose()
               for i in range(0,iter):
                   pre=np.dot(x,w)
                   loss=(pre -y)
                   gradent=np.dot(x_train.loss)/m
                   w=w-alpha*gradent
                   cost=1.0/2*m*np.sum(np.square(np.dot(x.np.transpose(w))-y))
                   print("第{}次梯度下降损失为：{}",format(i,round(cost,2)))
                   return w

               result=gradientDescent(1000,x_data,y_data,theta,0.01)
               y.pre=np.dot(x_data,result)
               print("线性回归模型 w.",result)
               plt.rc('font',family='Arial Uncode MS',size=14)
               plt.scatter(x[:,0],x[:,1],color='r',lable='训练数据')
               plt.plot(x[:,0],y_pre,color='g',lable='预测数据')
               plt.xlabel('x')
               plt.ylable('y')
               plt.title('线性回归预测（梯度下降）')
               plt.legend()
               plt.show()

运行结果

E:\YY\pythonProject\venv\Scripts\python.exe E:/YY/pythonProject/main.py
Traceback (most recent call last):
  File "E:/YY/pythonProject/main.py", line 2, in <module>
    import matplotlib.pyplot as plt
ImportError: No module named 'matplotlib'

Process finished with exit code 1

有没有大佬告诉我哪里错了