错误:
return K.categorical_crossentropy(y_true, y_pred, from_logits=from_logits)
C:\Users\selvaa\miniconda3\envs\tensorflow\lib\site-packages\tensorflow\python\keras\backend.py:4619 categorical_crossentropy
target.shape.assert_is_compatible_with(output.shape)
C:\Users\selvaa\miniconda3\envs\tensorflow\lib\site-packages\tensorflow\python\framework\tensor_shape.py:1128 assert_is_compatible_with
raise ValueError("Shapes %s and %s are incompatible" % (self, other))
ValueError: Shapes (None, 1) and (None, 151) are incompatible
我的模型:
x = np.array(x)
y = np.array(y)
x = x/255.0
model = Sequential()
model.add(Conv2D(3, (3,3), input_shape=(128,128,3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2,2)))
model.add(Flatten())
model.add(Dense(302, activation='relu'))
model.add(Dense(151, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit(x, y, batch_size=32, epochs=5, verbose=1, validation_split=0.1)
我试图训练一个模型来识别不同的口袋妖怪,我有两张151个口袋妖怪的图片用于我的数据集(正确标记和全部)。不知道我做错了什么
以下是打印x.shape和y.shape时发生的情况:
(301, 128, 128, 3) (301,)
使用loss^{} ,如下面的代码示例所示
丢失函数
tf.keras.losses.SparseCategoricalCrossEntropy
接受形状(n_samples,)
中的引用标签和形状(n_samples, n_classes)
中的预测标签,这将适用于您的数据。您不能使用categorical_crossentropy
,因为这要求您的标签是一个热编码的(请参阅答案的底部)另一个解决方案是在培训之前对标签进行热编码,例如使用函数^{} 。如果您使用这种方法,那么您可以使用
categorical_crossentropy
相关问题 更多 >
编程相关推荐