SVM python中predict_proba的提取概率

2017-09-19 163 views 0 likes

任何人都可以告诉我如何转换predict_proba SVM结果中的概率数据。结果如下例所示，但我想要一个只包含第二列并且没有括号的列表或数组。SVM python中predict_proba的提取概率

[[ 0.15941701 0.84058299] 
[ 0.12190033 0.87809967] 
[ 0.06293788 0.93706212] 
... 
[ 0.93175738 0.06824262]] 
<class 'numpy.ndarray'>

谢谢。

看到这个答案：[切片numpy的列（https://stackoverflow.com/questions/4455076/how-to-access-the-ith -column对的一-numpy的-多维阵列）。提取第二列= a [：，1]。 – blackplant

回答

你只需要使用numpy索引为多维数组。在这种情况下，您需要第一维（行）中的全部以及第二维（第二列）中的第二个元素。所以你用[:, 1]

a = np.array([ 
    [0.15941701, 0.84058299], 
    [0.12190033, 0.87809967], 
    [0.06293788, 0.93706212]]) 

a[:,1]

其中给出

[ 0.84058299 0.87809967 0.93706212]

2017-09-19 11:47:08

谢谢你们。这实际上是一个简单的任务。 –