2

我正在尝试在不使用 numpy 库的情况下手动计算最大化似然估计的协方差矩阵,但我似乎无法得到正确的答案。我试图通过这个公式:

协方差的最大似然估计

我知道我正在正确计算平均值。所以我实际计算协方差的部分一定有问题,但我不知道在哪里?这是我的代码:

mat = [[1,2,3],[4,6,8],[3,5,7]]

#now calc covariance for each element of the matrix
Cov = []
for j in range(len(means)):
    sum = 0
    covs = []
    for k in range(len(means)):
        for i in range(len(means)):
            sum += ((mat[i][j] - means[j]) * (mat[i][k] - means[k]))
        result  = sum/ len(means)
        covs.append(result)
    Cov.append(covs)
print(np.reshape(S,(3,3)))

这就是我得到的:

[[ 1.55555556  3.66666667  6.33333333]
[ 2.11111111  5.          8.66666667]
[ 2.66666667  6.33333333 11.        ]]

这是我应该得到的:

[[1.55555556 2.11111111 2.66666667]
[2.11111111 2.88888889 3.66666667]
[2.66666667 3.66666667 4.66666667]]
4

1 回答 1

3

您应该重置协方差矩阵的每个条目的总和,

    covs = []
    for k in range(len(means)):
        sum = 0
        for i in range(len(means)):
            sum += ((mat[i][j] - means[j]) * (mat[i][k] - means[k]))
        covariance  = sum/ len(means)
        covs.append(covariance)

你可以缩短一点

    covs = []
    for k in range(len(means)):
        terms = ( (mat[i][j] - means[j]) * (mat[i][k] - means[k]) for i in range(len(means)) )
        covariance  = sum(terms) / len(means)
        covs.append(covariance)

请务必清除工作区,以便它sum再次成为内置函数而不是数字。

于 2020-04-02T13:43:58.650 回答