/attention-mechanism-keras

attention mechanism in keras, like Dense and RNN...

Primary LanguagePython

attention-mechanism-keras

This repo implements the attention mechanism in keras.

To do list

  • Attention in Dense Network
  • Attention in RNN Network

Attention in Dense Network

Use attention mechanism to get which feature shuold be paid more attention. A Dense layer is used to get the probs.Here is the probs plot.
image

Attention in RNN Network

Use attention mechanism to get which timesteps shuold be paid more attention( of cause, you can switch to which features shuold be paid more attention).Here is the probs plot.
image