AI engineering glossary · NumPy pandas visualization
What is argmax?
argmax (also called np.argmax, argmin): A function that returns the position of the largest value in an array rather than the value itself. With an axis argument it does this per row or column. In machine learning it turns a vector of class scores into a predicted class label.
argmax: an example
np.argmax([0.1, 0.7, 0.2]) returns 1, so class 1 is the prediction.
Related terms
- argsort
- Reduction
- Axis
- Classification
More NumPy pandas visualization terms
Where you learn it
argmax is part of the AI Engineering Program, which teaches it with notes, a hands-on lab and an exam. The first lesson is free.