我如何获得keras模型以输出概率作为输出?

时间:2019-04-16 17:31:50

标签: keras

如何使keras模型显示概率,例如jq -cr '([.ps] | @tsv) + "\t" + tostring' 而不是[0.3,0.1,0.6]。 keras模型如下所示。我在最后一层使用softmax,但是我不确定我在做什么。

[0.,0.,1.]

这是我运行model = Sequential() model.add(Conv2D(32,(3, 3),activation='relu',input_shape=train_tensors[0].shape)) model.add(MaxPooling2D(pool_size=(2, 2), strides=None, padding='same')) model.add(Dropout(0.2)) model.add(Conv2D(64, (3,3), activation='relu')) model.add(MaxPooling2D(pool_size=(2, 2), strides=None, padding='same')) model.add(Dropout(0.2)) model.add(Conv2D(128, (3,3), activation='relu')) model.add(MaxPooling2D(pool_size=(2, 2), strides=None, padding='same')) model.add(Dropout(0.2)) model.add(Flatten()) model.add(Dense(64, activation='relu')) model.add(Dropout(0.3)) model.add(Dense(133, activation='softmax')) model.summary() model.compile(optimizer='rmsprop', loss='categorical_crossentropy', metrics=['accuracy']) model.fit(train_tensors, train_targets, validation_data=(valid_tensors, valid_targets), epochs=epochs, batch_size=20, callbacks=[checkpointer], verbose=1)

时的输出结果
model.predict(img)

0 个答案:

没有答案