Модель построена с использованием tf.keras, ссылка на записную книжку: введите здесь описание ссылки
Я использую этот ноутбук в google colab.