Привет! Меня зовут Макар Бадерко, я 12-летний разработчик машинного обучения. Сегодня мы попытаемся занять первое место в одном из самых популярных соревнований Kaggle по компьютерному зрению — Digit Recognition со знаменитым набором данных MNIST. Это будет не очень сложно, так что приступим!
Я рекомендую вам запустить этот код в Kaggle Notebooks.
Для начала давайте импортируем несколько необходимых библиотек:
После этого давайте загрузим наши данные и посмотрим на образец отправки:
Мы хотели бы поближе рассмотреть некоторые из наших фотографий.
На этом шаге вам необходимо загрузить полный набор данных MNIST с этого веб-сайта:
https://www.kaggle.com/oddrationale/mnist-in-csv
Здесь вы создадите генератор данных изображения, который сделает данные лучше всего подходящими для нашей модели.
Здесь мы создадим модель TensorFlow:
Вы можете найти значение всех этих параметров здесь:
https://www.tensorflow.org/js/guide/models_and_layers
Давайте подгоним нашу модель к данным MNIST.
Я буду делать прогнозы на тестовом наборе данных Kaggle.
И сохраните их в файл CSV
Спасибо за прочтение, подписывайтесь на меня на Medium! :)