/mnist

The MNIST database of handwritten digits, available from this page, has a training set of 60,000 examples, and a test set of 10,000 examples. It is a subset of a larger set available from NIST. The digits have been size-normalized and centered in a fixed-size image.

Primary LanguagePython

Описание скриптов:

  • train.py - загрузка данных из базы данных keras.mnist, создание нейросети и обучение на этих данных.
  • test.py - загрузка обученной модели и проверка на картинках из data (0 - 9.png), выдача результата (result.json) загрузка данных из базы данных и создание confusion-матрицы.
  • confusion.py - загрузка обученной модели из model/model.h5 и создание confusion-матрицы.

Создание и активация виртуальной среды:

  • python3 -m venv venv
  • source venv/bin/activate

Установка зависимостей:

  • pip install -r requirements.txt

Запуск скриптов:

  • python3 train.py
  • python3 test.py
  • python3 confusion.py