2017-07-06 8 views
1

私は画像をいくつかのカテゴリに分類するケラスモデルを持っています。モデルが画像を分類するカテゴリの数を印刷するにはどうすればよいですか?カテゴリの数、ケラス

+1

あなたの問題を解決しようとした方法/コードを投稿して投稿を改善してください。 –

+1

モデルの出力レイヤーにはいくつのニューロンがありますか? –

+0

出力ニューロンの数によって、あなたが持っているカテゴリの数が決まります。 – Lamda

答えて

1

Kerasのmodel.summary()を使用すると、モデルの各レイヤーの情報を順番に確認できます(下の画像を参照)。 enter image description here

出力レイヤーの最後のレイヤーでは、出力シェイプが(None、10)であることがわかります。このレイヤーには、10個のクラスまたは10個のカテゴリが含まれています。 (バッチサイズを意味するものは何も心配しないでください)。