argmax(),argmin()是numpy模組中的函式:
直接舉例說明:
import numpy as np
y = np.array([[
0.1,
0.98
,0.69],
[0.45
,0.78
,0.99]]
)# 隨便定義乙個二維陣列
y輸出:
array([[
0.1,
0.98
,0.69],
[0.45
,0.78
,0.99]]
)
axis = 1,表示對行操作,argmax是取一行中最大值對應的下標
a = y.argmax(axis =1)
a輸出:
array([1
,2], dtype=int64)
axis = 0,表示對列操作,argmax是取一列中最大值對應的下標
a = y.argmax(axis =1)
a輸出:
array([1
,0,1
], dtype=int64)
argmax()函式可以用在多分類任務的輸出階段,對每個樣本輸出的**值直接取最大值對應的下標,很是方便
至於argmin()函式,是取相應的最下元素,這裡就不多說!
argmax和argmin函式的使用
argmax a,axis none,out none a 表示array axis 表示指定的軸,預設是none,表示把array平鋪,out 預設為none,如果指定,那麼返回的結果會插入其中這個函式返回的是沿軸axis最大值的下標 索引值 1 如果沒有指定axis的值,array就會被平鋪成乙...
numpy裡面的argmax函式
函式原型 def argmax a,axis none,out none a 輸入array axis 為0代表列方向,為1代表行方向 out 結果寫到這個array裡面 1 從簡單的一維開始 import numpy as np a np.array 3,1,2,4,6,1 bb np.argma...
numpy裡面的argmax函式
numpy裡面的argmax函式 函式原型 def argmax a,axis none,out none a 輸入array axis 為0代表列方向,為1代表行方向 out 結果寫到這個array裡面 例子 import numpy as np a np.array 2,4,6,1 1,5,2,...