Predictions - index with maximum value on second axis?
Fine-tuning a model with the Trainer API - Hugging Face CourseIt says on the above link:As you can see,predictionsis a two-dimensional array with shape 408 x 2 (408 being the number of elements in the dataset we used). Those are the logits for each element of the dataset we passed topredict()(as you saw in theprevious chapter, all Transformer models return logits).To transform them into predictions that we can compare to our labels, we need to take the index with the maximum value on the second axis:what does it mean that we need to take the index with the maximum value on the second axis?What two axes are we talking about?