Python中numpy函数的使用

xiaoxiao2021-02-28  47

一.求线性方程的斜率

①最小二乘法

import numpy as np from numpy.linalg import inv #逆矩阵 from numpy import dot #矩阵乘 from numpy import mat #矩阵 #y=2x 求线性方程的斜率 X = mat([1, 2, 3]).reshape(3, 1) #reshape 的作用就是修改矩阵的维度 print(X) Y = 2*X # theta = (X'X)-1X'Y theta = dot(dot(inv(dot(X.T, X)), X.T), Y) print(theta)

②梯度下降算法

import numpy as np from numpy.linalg import inv #逆矩阵 from numpy import dot #矩阵乘 from numpy import mat #矩阵 #y=2x 求线性方程的斜率 X = mat([1, 2, 3]).reshape(3, 1) #reshape 的作用就是修改矩阵的维度 Y = 2*X # theta = (X'X)-1X'Y #theta = dot(dot(inv(dot(X.T, X)), X.T), Y) #theta = theta - alpha*(theta*X-Y)*X 梯度下降算法 theta = 1.0 alpha = 0.1 for i in range(100): theta = theta + np.sum(alpha * (Y - dot(X, theta))*X.reshape(1, 3))/3. print(theta)
转载请注明原文地址: https://www.6miu.com/read-2449986.html

最新回复(0)