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%