capsule_attention

The source code with paper: HIGH-FREQUENCY VOLTAGE SIGNAL FAULT DIAGNOSIS BASED ON MULTI-CHANNEL HIERARCHICAL STRUCTURE CAPSULE-ATTENTION NETWORK

1. usage

python main.py

2. Install Dependencies(Recomend Virtualenv)

  • keras >= 2.1
  • tensorflow >= 2.1