Изменить код , чтобы улучшить точность тестовой выборке для задачи классификации изображений

Нужно улучшить код таким образом чтобы модель давала лучшие результаты на тестовой выборке желательно от 0.78 .Или изменить сам подход если есть лучшие варианты.Сам код прикрепляю в файлы а также ссылки на тестовые и тренировочные данные.
train: drive.google.com/file/d/1...
test: drive.google.com/drive/fo...