如何在keras中以百分比显示预测结果？

Question

在 keras 中始终以一种热门格式输出的结果如下所示：

[[ 0.  0.  0.  0. 0. 0. 1. 0]]
[[ 0.  1.  0.  0. 0. 0. 0. 0]]

但我想对每个分类都有一个概率百分比，例如

[[ 0.54  0.80  0.34  0.041]]

下面是我的简单代码

import numpy as np
from keras.models import load_model

from new_get_data_for_test import load_my_data

try :
   X_test = load_my_data()

   model = load_model('/home/istiyak/development/pythonProject/my_save_model.h5')

   output = model.predict(X_test)

   print("After pridiction our output is : ")
   print (output)
   print(np.argmax(output, axis=1))

except Exception as e :
   print ("Some Exception occurs")

Answer 1

使用predict_proba。来自文档：

predict_proba(self, x, batch_size=32, verbose=1)

逐批为输入样本生成类别概率预测。

返回一个概率预测的 Numpy 数组。