3 references to argmax
Microsoft.ML.OnnxTransformerTest (3)
OnnxTransformTests.cs (3)
453
Assert.Equal(1, predictions[0].
argmax
[0]);
454
Assert.Equal(0, predictions[1].
argmax
[0]);
455
Assert.Equal(2, predictions[2].
argmax
[0]);