本教程基于pytorch实现堆叠自编码神经网络,包含网络模型构造、训练、测试的详细步骤。其中包含训练与测试数据(.mat文件)、模型(AE_ModelConstruction.py、AE_Train.py)以及测试例子(AE_Test.py)等重要内容,数据集涉及风粉混合物温度等14个变量。神经网络的层数和每层神经元个数无固定,可以根据用户输入值来构造,方便调试。初始化时autoencoder类有三个参数:网络输入值、SAE编码过程的层数(编码、解码过程层数相同)和是否添加BN层。支持原有本义、保留专业术语的标题改写,据此,修改了标题内容,满足搜索引擎优化的要求。
暂无评论