1.2 正规方程的推导及Python验证

最新推荐文章于 2022-11-11 00:42:46 发布

程序大猩猩

最新推荐文章于 2022-11-11 00:42:46 发布

阅读量552

点赞数

分类专栏： cs229(吴恩达机器学习) 文章标签：线性回归正规方程机器学习算法

本文链接：https://blog.csdn.net/weixin_43864473/article/details/86583559

版权

cs229(吴恩达机器学习) 专栏收录该内容

11 篇文章 3 订阅

订阅专栏

在这里插入图片描述

import numpy as np
#X矩阵的维度10*2
m=100
n=2
#定义输入矩阵、参数、输出列向量
X=np.random.randint(1,m,m*n).reshape(m,n)
#这里的参数数量和n相等就行
pre_theta=np.array([1.872,100.221]).T
y=np.dot(X,pre_theta)
#求秩，需要判断下是否为奇异矩阵
A=np.dot(X.T,X)
B=np.dot(X.T,y)
rank=np.linalg.matrix_rank(A)
#不是直接求解就行，是的话求最小二乘解
if rank==min(m,n):
    theta=np.linalg.tensorsolve(A,B)
else:
    theta=np.linalg.lstsq(A,B)
print(theta)