attention_rnn

A little project made for recruitment. My task was to implement a recurrent neural network with an attention module for a classification problem. Dataset consists of sentences in English. There are seven labels representing intentions.

evaluate.py

Demo script. A trained network weights file is needed to run.

train.py

The file with the training function.

data_utils.py

Some utility functions to ease data preparation.

model.py

The model definition.

Training results

Class labels:

• 0 - AddToPlaylist
• 1 - BookRestaurant
• 2 - GetWeather
• 3 - PlayMusic
• 4 - RateBook
• 5 - SearchCreativeWork
• 6 - SearchScreeningEvent

Dataset source