pub fn argmax(probabilities: &[f64]) -> usize
The index of the largest probability. Ties go to the lower index.