SVM python中predict_proba的提取概率

2017-09-19 165 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

謝謝你們。這實際上是一個簡單的任務。 –

相關問題