基于胶囊网络的齿轮箱故障诊断

问题遇到的现象和发生背景

请问想用胶囊网络做一下齿轮或者轴承的故障诊断，有没有胶囊网络相关代码推荐参考呢