如何在python中使用权重矩阵的最小二乘？

发布时间：2020-12-16 23:38:25 所属栏目：Python 来源：网络整理

导读：我知道如何使用 python通过最小二乘法解决A.X = B：例： A=[[1,1,1],[1,0]]B=[1,1]X=numpy.linalg.lstsq(A,B)print X[0]# [ 5.00000000e-01 5.00000000e-01 -1.66533454e-16 -1.11022302e-16] 但是如果用权重矩阵不是身份来解决这个相同的等式呢： A.X = B

我知道如何使用 python通过最小二乘法解决A.X = B：

例：

A=[[1,1,1],[1,0]]
B=[1,1]
X=numpy.linalg.lstsq(A,B)
print X[0]
# [  5.00000000e-01   5.00000000e-01  -1.66533454e-16  -1.11022302e-16]

但是如果用权重矩阵不是身份来解决这个相同的等式呢：

A.X = B (W)

例：

A=[[1,1]
W=[1,2,3,4,5]

谢谢提前,

解决方法

我不知道你如何定义你的体重,但你可以尝试这个,如果合适的话：

import numpy as np
A=np.array([[1,0]])
B = np.array([1,1])
W = np.array([1,5])
Aw = A * np.sqrt(W[:,np.newaxis])
Bw = B * np.sqrt(W)
X = np.linalg.lstsq(Aw,Bw)

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!