学习
Arthur_fuck
这个作者很懒,什么都没留下…
展开
-
CWRU数据集12K重叠采样
之前所用的代码是在tensorflow的基础上强改的,还是存在一些问题,比如样本个数过少之类缺点。最近自己在网上缝缝补补,总算是拼凑出一个勉强可以一用的代码。供大家参考,如果有不足之处,请大家批评指正,欢迎讨论交流。Tips:这里的label不能从0开始,否则进网络的话,交叉熵计算会出现问题,好像和torch的底层代码有关系,我是没搞懂的。原创 2023-03-28 18:21:07 · 324 阅读 · 1 评论 -
西储大学(CWRU)轴承数据集的数据读取与划分
由于课题原因,最近在学习torch对于轴承故障检测的相关知识,但第一步读取数据以及数据的划分就难到了我,在网上查找相关资料,也没有完整的代码,于是只能东拼西凑,修修改改,最后勉强凑出来一个可以一用的代码,放在这里保存一下。原创 2022-12-09 10:38:43 · 2023 阅读 · 1 评论 -
关于if --name-- == ‘--main--‘判断的运用
在上次的网络学习中,觉得在推理过程中需要重新定义网络,觉得过于繁琐,便想到能不能将train中的网络定义import到test函数中,但在运行测试函数的过程中发现,导入训练函数会导致网络重新训练,于是上网查阅资料,发现了问题所在并找到了解决方法,也就是 if --name-- == ‘–main–’。if --name-- == ‘–main–’:这个判断可以用于判断文件是作为一个脚本文件运行,还是作为一个包导入到新的文件中。如果该函数为真,则作为脚本文件运行;为假,则作为包导入。原创 2022-12-02 21:56:37 · 798 阅读 · 0 评论 -
基于Pytorch的CNN网络对Cifar10数据集进行分类
本博客仅为学习经历记录。原创 2022-11-29 20:56:10 · 1736 阅读 · 6 评论