- 博客(2)
- 收藏
- 关注
RSS订阅原创 比较pytorch与keras训练模型的不同并且在Pytorch中使用一维卷积神经网络对一维连续数据分类
卷积神经网络对于周期性的一维信号分类效果是比较好的，本人做电机故障检测，需要对不同故障的轴承振动信号进行分类。之前都是用keras搭建神经网络，确实很方便，使用fit（）函数训练模型确实很省事。但是fit（）函数不太灵活，如果想对网络中间层输出特征操作的话好像是不行的（博主也是初学者，如果有大佬知道keras或者tensorflow如何对中间输出特征操作的话还望不吝赐教）。所有，最近看了一下pytorch，并用pytorch搭建了一维卷积神经网络对凯斯西储大学数据集进行分类。在过程中...
2022-05-10 17:29:45
3732
13
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人