5 references to argmax
Microsoft.ML.OnnxTransformerTest (5)
OnnxTransformTests.cs (5)
822
new[] { nameof(PredictionWithCustomShape.
argmax
) }, new[] { nameof(InputWithCustomShape.input) },
829
nameof(PredictionWithCustomShape.
argmax
), nameof(InputWithCustomShape.input),
858
Assert.Equal(transformedDataPoints[i].
argmax
, expectedResults[i]);
984
var pipeline = ML.Transforms.ApplyOnnxModel(nameof(PredictionWithCustomShape.
argmax
),
1022
Assert.Equal(transformedDataPoints[i].
argmax
, expectedResults[i]);