小编use*_*147的帖子

Keras softmax 激活,category_crossentropy 损失。但输出不是0、1

我只用很少的数据训练了一个 epoch 的 CNN 模型。我使用 Keras 2.05。

这是 CNN 模型的(部分)最后 2 层number_outputs = 201。训练数据输出是一种热编码 201 输出。

model.add(Dense(200, activation='relu', name='full_2'))
model.add(Dense(40, activation='relu',  name='full_3'))
model.add(Dense(number_outputs, activation='softmax'))

model.compile(loss='categorical_crossentropy', optimizer=adam, metrics=['accuracy'])
Run Code Online (Sandbox Code Playgroud)

模型保存为 h5 文件。然后,保存的模式将加载与上面相同的模型。batch_image是一个图像文件。

prediction = loaded_model.predict(batch_image, batch_size=1)
Run Code Online (Sandbox Code Playgroud)

我得到这样的预测:

ndarray: [[ 0.00498065  0.00497852  0.00498095  0.00496987  0.00497506  0.00496112
   0.00497585  0.00496474  0.00496769  0.0049708   0.00497027  0.00496049
   0.00496767  0.00498348  0.00497927  0.00497842  0.00497095  0.00496493
   0.00498282  0.00497441  0.00497477  0.00498019  0.00497417  0.00497654
   0.00498381  0.00497481  0.00497533  0.00497961  0.00498793  0.00496556
   0.0049665   0.00498809  0.00498689  0.00497886  0.00498933  0.00498056
Run Code Online (Sandbox Code Playgroud)

问题:

  1. 预测数组应该是1, …

keras softmax

5
推荐指数
1
解决办法
2万
查看次数

标签 统计

keras ×1

softmax ×1