我有一个CNN,输出维度为[无,10]
这是一个多标签问题,其中输出表示x可能属于的可能类别。 (例如,图像可以归类为猫暗等等)
…
在Keras:
#you model here -- last layer: model.add(Dense(10)) model.add(Activation('sigmoid')) model.compile(loss='categorical_crossentropy', optimizer="adam",metrics=['accuracy'])
Keras中的直接替代方法是在输出层中使用sigmoid激活,并将binary_crossentropy用作成本函数。
net.add(Dense(..., activation='sigmoid')) net.compile(optimizer, loss='binary_crossentropy')
看一看 https://github.com/keras-team/keras/issues/741