Matlab不同类的协方差矩阵计算

Question

我有2个不同的文件，其中一个是输入矩阵(X)，有3823*63个元素(3823个输入和63个特征)，另一个是类向量(R)，有3823*1个元素元素；这些元素的值从 0 到 9(有 10 个类)。

我必须计算每个类的协方差矩阵。到目前为止，我只能计算具有如此多嵌套循环的每个类的平均向量。然而，它让我脑死亡。

还有其他简单的方法吗？

<小时/>

这里有适合我目的的代码(感谢 Sam Roberts):

xTra = importdata('optdigits.tra');
xTra = xTra(:,2:64); % first column's inputs are all zero

rTra = importdata('optdigits.tra');
rTra = rTra(:,65); % classes of the data

c = numel(unique(rTra));

for i = 1:c
    rTrai = (rTra==i-1); % Get indices of the elements from the ith class
    meanvect{i} = mean(xTra(rTrai,:)); % Calculate their mean
    covmat{i} = cov(xTra(rTrai,:)); % Calculate their covariance
end

Answer 1

这能满足您的需要吗？

X = rand(3263,63);
R = randi(10,3263,1)-1;

numClasses = numel(unique(R));

for i = 1:numClasses
    Ri = (R==i); % Get indices of the elements from the ith class
    meanvect{i} = mean(X(Ri,:)); % Calculate their mean
    covmat{i} = cov(X(Ri,:)); % Calculate their covariance
end

此代码循环遍历每个类，选择 R 中与该类的观测值相对应的行，然后从 X 中获取相同的行并计算它们的均值和协方差。它将它们存储在元胞数组中，因此您可以像这样访问结果:

% Display the mean vector of class 1
meanvect{1}

% Display the covariance matrix of class 2
covmat{2}

希望有帮助!