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 ·
887 阅读 ·
0 评论