【发布时间】:2017-04-13 06:16:01
【问题描述】:
我训练了一个猫狗分类模型,做的时候不知道怎么得到每个类别的概率
【问题讨论】:
-
你能说得具体一点吗,伙计?
我训练了一个猫狗分类模型,做的时候不知道怎么得到每个类别的概率
【问题讨论】:
您只需要在网络顶部有两个神经元。 所以在卷积层之后:
model = Sequential()
model.add(Convolution2D(32, 3, 3, input_shape=(3,img_rows,img_cols),dim_ordering="th"))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dense(2))
model.add(Activation('sigmoid'))
model.compile(loss='categorical_crossentropy')
【讨论】: