Film review classification

Решение задачи классификации отзывов на фильмы на положительные и отрицательные с помощью рекуррентных нейронных сетей

1. Задача:

Для набора отзывов на фильмы из Internet Movie Database (IMDB) необходимо научиться классифицировать отзывы на положительные и отрицательные

2. Технологии:

• Python 3
• Pytorch
• Numpy
• Torchtext
• Google Colab

3. Модель:

• Embedding-слой для представления индексов в отзыве в качестве векторов
• RNN/GRU-ячейки для обработки текста (рекуррентные нейронные сети)
• Linear + SoftMax слои для классификации текста

4. Результаты:

Точность предсказаний (метрика accuracy): 86%