多元正态分布python平均向量的计算

2024-10-06 08:05:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我很难将多元高斯分布拟合到我的数据集,更具体地说,找到一个平均向量(或多个平均向量)。我的数据集是一个nx8矩阵,目前我正在使用以下代码:

muVector = np.mean(Xtrain, axis=0)其中Xtrain是我的训练数据集。在

对于协方差,我使用任意方差值(.5)构建它,并执行以下操作:

covariance = np.dot(.5, np.eye(N,N)其中N是观察数。在

但当我构造φ矩阵时,我得到的都是零。这是我的代码:

muVector = np.mean(Xtrain, axis=0)
# get covariance matrix from Xtrain
cov = np.dot(var, np.eye(N,N))
cov = np.linalg.inv(cov)  

# build Xtrain Phi
Phi = np.ones((N,M))
for row in range(N):
  temp = Xtrain[row,:] - muVector
  temp.shape = (1,M)
  temp = np.dot((-.5), temp)
  temp = np.dot(temp, cov)
  temp = np.dot(temp, (Xtrain[row,:] - muVector))
  Phi[row,:] = np.exp(temp)

感谢任何帮助。我想我得用np.随机.多元正态分布()? 但我不知道在这种情况下如何使用它。在


Tags: 数据代码np矩阵meancov向量temp
1条回答
网友
1楼 · 发布于 2024-10-06 08:05:55

我相信“Phi”是指你想要估计的概率密度函数(pdf)。在这种情况下,协方差矩阵应为MxM,输出功率因数为Nx1:

# -*- coding: utf-8 -*-

import numpy as np

N = 1024
M = 8
var = 0.5

# Creating a Xtrain NxM observation matrix.
# Its muVector is [0, 1, 2, 3, 4, 5, 6, 7] and the variance for all
# independent random variables is 0.5.
Xtrain = np.random.multivariate_normal(np.arange(8), np.eye(8,8)*var, N)

# Estimating the mean vector.
muVector = np.mean(Xtrain, axis=0)

# Creating the estimated covariance matrix and its inverse.
cov = np.eye(M,M)*var
inv_cov = np.linalg.inv(cov)  

# Normalization factor from the pdf.
norm_factor = 1/np.sqrt((2*np.pi)**M * np.linalg.det(cov))

# Estimating the pdf.
Phi = np.ones((N,1))
for row in range(N):
    temp = Xtrain[row,:] - muVector
    temp.shape = (1,M)
    temp = np.dot(-0.5*temp, inv_cov)
    temp = np.dot(temp, (Xtrain[row,:] - muVector))
    Phi[row] = norm_factor*np.exp(temp)

或者,您可以使用scipy.stats.multivariate_normal中的pdf方法:

^{pr2}$

Phi和{}输出数组将相等。在

相关问题 更多 >