Сверточная нейросеть для определения жанра постера

Код для обучения модели сверточной нейросети представлен на Google Colab https://colab.research.google.com/drive/1EGf6XWZa-nhl4eUmzdSyFbGUCzTLBL3W#scrollTo=U4XsAodsYAFO

Файл movies_dataset.csv

содержит данные о постере, такие как: уникальный номер, дата выхода, название фильма, рейтинг на сайте imdb, ссылку на фильм на imdb, список жанров фильма.

Папка movies_posters

содержит 10 папок

[Action, Comedy, Documentary, Drama, Family, Horror, Mystery, Romance, Sience Fiction, Thriller]

с файлами формата .jpg, подписи файлов соответствуют жанру фильма соответствующего постера.

Файлы с кодом, который был применен для подготовки данных

Файл gen.py

искусственно генерирует картинки из имеющиехся в папке, если в ней существенно меньше 5000 постеров.

Файл papki.py

содержит код создания папок и уменьшения размеров изначальных картинок в 5 раз.