基于注意机制的小样本故障诊断模型及pytorch实现
该故障诊断模型基于注意机制,并提供了pytorch实现代码。使用环境需要pytorch版本号为1.10.0及以上,Python版本号为3.8及以上,CUDA版本号为10.2及以上。代码可以直接运行,适用于小样本情境下的故障诊断任务。
下载地址
用户评论
得注意的一点是,文章并未针对样本数量进行深入研究;
者对模型进行了详细的介绍和分析;
章将注意机制技术应用到故障诊断领域,具有创新意义;
文使用的实验数据组织合理,可信度较高;
究对于实际工业应用有一定的参考价值;