问题描述:
在从二分类做到五分类的时候,报错如下:
Error when checking target: expected activation_9 to have shape (2,) but got array with shape (5,)
解决方案
????y1?=?keras.layers.core.Dense(5)(y1)#五分类而非二分类
y_train = keras.utils.to_categorical(y_train,num_classes=5)
y_test = keras.utils.to_categorical(y_test,num_classes=5)
y1 = keras.layers.core.Dense(5)(y1)#五分类而非二分类
所以需要更改两个地方,一个是num_classes的值改为5,同时训练模型最后一层的密集层也要改为5。
另外以下两点也要特别注意。标签命名时候也有一定的规矩。
- label的取值必须从0开始;
- label必须是连续的整数(不能间断);
总结起来就是:label的取值必须是从0开始的连续整数。 (假如有5类,那么label的取值1,2,3,4,5或者0,1,2,4,5都是不行的)。
从二分类转变为多分类主要改这两个地方的代码。
运行结果:
参考链接:
https://stackoverflow.com/questions/49392972/error-when-checking-target-expected-dense-3-to-have-shape-3-but-got-array-wi
https://blog.csdn.net/qq295456059/article/details/77264081
|