-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
你可能會更具體一點,交配? –