Advertisement

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)
    
    
      
      
      
      
      
      
      
      
      
      
      
      
      
      
    

具体实验数据的呈现形式如右图所示:

![在这里插入图片描述](https://

全部评论 (0)

还没有任何评论哟~