-1
我训练一只猫与狗分类模型,当我这样做,我不知道如何让每个类别的概率如何预测类的概率?
我训练一只猫与狗分类模型,当我这样做,我不知道如何让每个类别的概率如何预测类的概率?
你只需要在顶部两个神经元你的网络。 所以卷积层之后:
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')
谢谢,我试试 – KyRie
你可能会更具体一点,交配? –