回答編集履歴
1
fixed context
test
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
TensorFlowのリファレンス[tf.keras.utils.image_dataset_from_directory](https://www.tensorflow.org/api_docs/python/tf/keras/utils/image_dataset_from_directory)をちゃんと読まれましたか？
|
|
2
2
|
|
|
3
|
-
この関数にはデフォルト引数`label_mode='int'`が設定されており，ラベル
|
|
3
|
+
この関数にはデフォルト引数`label_mode='int'`が設定されており，ラベルを整数値とした損失関数`sparse_categorical_crossentropy`用に設定されています．与えたデータが`(None, 1)`であるのに対して，ネットワークの出力は`(None, 4)`であることの不整合性からエラーが起きています．
|
|
4
4
|
|
|
5
5
|
現状のコードは損失関数`categorical_crossentropy`用にOne-Hot Encodingされたラベルでなくてはならないはずです．したがって，引数`label_mode='categorical'`を指定しましょう．
|