线性判别分析(LDA)的Pytorch实现

一、概述
结合代码给出LDA算法步骤，并在本文最后给出Python的全部代码。
二、代码实现
1、生成数据集

##################################生成数据集####################################
#类别1
x1=torch.linspace(0,1,10)
xx1=3*x1-4+torch.normal(0,1,(1,10))
#类别2
x2=torch.linspace(-1,0,10)
xx2=3*x2+2+torch.normal(0,1,(1,10))
#绘图
plt.scatter(x1,xx1,30)
plt.scatter(x2,xx2,30)

plt.plot()
plt.xlim(-2, 2)
plt.ylim(-8, 6)

运行结果如下图所示：

2、计算类内散度矩阵

#计算类内散度矩阵
Cov1=np.cov(x1,xx1)
Cov2=np.cov(x2,xx2