To get the index of the max value in a Numpy array, use np.argmax. Below is an example.
np.argmax
This can be helpful to get the highest probability in an array of probabilities.
array-to-latex: Turn a Numpy Array into Latex
unyt: Manipulate and Convert Units in NumPy Arrays
NumPy.take: Take Elements From an Array
Your email address will not be published. Required fields are marked *
Name
Email
Website
Save my name, email, and website in this browser for the next time I comment.
Δ