Skip to content

sk409/MNIST

Repository files navigation

自作機械学習ライブラリでMNISTの分類器を作成

自作の機械学習ライブラリで分類器の訓練を行いました。

mnist_main.pyで実際に学習を行い、学習過程のログを確認することができます。

mnist_test.pyで今回学習させたモデルにMNIST画像を読み込ませ何の数字であるかを予測させることができます。


分類期の性能

訓練データに対する精度: 99%

検証データに対する精度: 97%

テストデータに対する精度: 97%

学習させたモデル

訓練終了後にmodelsフォルダ直下に訓練させたモデルが保存されます。

modelsフォルダにあるbest_modelフォルダには今までで一番性能の良かったモデルを格納してあります。

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages