3 references to argmax
Microsoft.ML.OnnxTransformerTest (3)
OnnxTransformTests.cs (3)
541
Assert.Equal(1, predictions[0].
argmax
[0]);
542
Assert.Equal(0, predictions[1].
argmax
[0]);
543
Assert.Equal(2, predictions[2].
argmax
[0]);