【深層学習、keras】mnistの数字の回帰モデルをkerasで作る

kerasでmnistの数字の画像を入力したら、その数字を出力するような回帰モデルを
作ってみます。
0と書いてある画像を入力したら、0を出力し、
9と書いてある画像を入力したら、9と出力するようにモデルを作成します。
以下はイメージ図

f:id:ttt242242:20190331135232j:plain

ちなみにCNNは使いません!

データセット

有名なmnistを用います。
手書き文字の認識用データセットです。
0〜9までの手書きの数字の画像が、学習用、テスト用でそれぞれ60000枚、10000枚用意されています。
特別ダウンロードする必要はありません。
下のプログラムを実行すれば、自動的にダウンロードされます。

プログラム

結果

プログラムにある通り、いくつかtestデータを取り出して、結果を簡易的に見てみます。

今回は過学習とかは特に考慮していませんが、ある程度うまくいってそうです。
あと、正解データも正規化したほうが良かったのですが、
今回はしませんでした。

終わりに

さすがにkerasを使うとシンプルに記述できます。
kerasでもう少しいろいろやってみたいです。

コメント

タイトルとURLをコピーしました