argmax 函式和argmin 函式

2021-09-28 14:33:33 字數 759 閱讀 1191

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,...