MNISTのサンプルコード

モデル

  • CNN
  • CNN + Autoencoder

学習

python train.py

検証

model_pathに学習済みモデルを指定することで認識率を返す

python predict.py

環境

  • Python 3.6.4
  • Keras 2.1.2
  • tensorflow 1.4.1 (backend)