Error when checking target: expected softmax to have shape (10,) but got array with shape (1,)
出现这个问题是因为loss function与label 的encode格式不匹配
第一种情况:
loss=‘sparse_categorical_crossentropy’
label请用noe-hot编码
第二种情况:
loss=‘categorical_crossentropy’
label是整型就可以了。
其中CIFAR-10的label编码默认为整型,对应的loss function 就应该设为categorical_crossentropy;
但如果要用sparse_categorical_cr
原创
2020-06-29 22:19:00 ·
828 阅读 ·
0 评论