如何在keras中显示预测的百分比结果?

时间:2017-01-02 14:07:07

标签: python-2.7 machine-learning deep-learning keras conv-neural-network

输出的结果总是以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")

先谢谢。

1 个答案:

答案 0 :(得分:2)

使用predict_proba。来自文档:

  

openjdk-jre-8-headless

     

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

     

返回Numpy概率预测数组。