【问题标题】:How to predict class probabilities?如何预测类别概率?
【发布时间】:2017-04-13 06:16:01
【问题描述】:

我训练了一个猫狗分类模型,做的时候不知道怎么得到每个类别的概率

【问题讨论】:

  • 你能说得具体一点吗,伙计?

标签: python keras predict


【解决方案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')

【讨论】:

    猜你喜欢
    • 2018-12-24
    • 2016-12-04
    • 2019-09-16
    • 2018-04-19
    • 2020-08-25
    • 2016-05-23
    • 2018-10-12
    • 2013-01-26
    相关资源
    最近更新 更多