5 references to argmax
Microsoft.ML.OnnxTransformerTest (5)
OnnxTransformTests.cs (5)
910
new[] { nameof(PredictionWithCustomShape.
argmax
) }, new[] { nameof(InputWithCustomShape.input) },
917
nameof(PredictionWithCustomShape.
argmax
), nameof(InputWithCustomShape.input),
946
Assert.Equal(transformedDataPoints[i].
argmax
, expectedResults[i]);
1072
var pipeline = ML.Transforms.ApplyOnnxModel(nameof(PredictionWithCustomShape.
argmax
),
1110
Assert.Equal(transformedDataPoints[i].
argmax
, expectedResults[i]);