基于pytorch实现的堆叠自编码神经网络,包含网络模型构造、训练、测试
基于pytorch实现的堆叠自编码神经网络,包含网络模型构造、训练、测试主要包含训练与测试数据(.mat文件)、模型(AE_ModelConstruction.py、AE_Train.py)以及测试例子(AE_Test.py)其中ae_D_temp为训练数据,ae_Kobs3_temp为正常测试数据,ae_ver_temp为磨煤机堵煤故障数据,数据集包含风粉混合物温度等14个变量在程序中神经网络的层数和每层神经元个数没有固定,可根据使用者的输入值来构造神经网络,方便调试autoencoder类在初始化时有三个参数,第一个是网络输入值,第二个是SAE编码过程的层数(编码、解码过程层数相同),第三个是是否添加BN层这里为了构造方便,给每层神经元的个数与层数建立一个关系:第一层神经元的个数为2^(layer数+2),之后逐层为上一层的1/2
文件列表
SAE_pytorch-master.zip
(预估有个13文件)
SAE_pytorch-master
ae_D0_temp.mat
9.53MB
ae_D2_temp.mat
937KB
ae_withBN_3.pkl
29KB
ae_Kobs3_temp.mat
658KB
ae_withBN_2.pkl
20KB
Loss_withBN_2.npy
103KB
AE_Train.py
3KB
AE_Test.py
5KB
AE_ModelConstruction.py
4KB
用户评论