5 references to PredictionWithCustomShape
Microsoft.ML.OnnxTransformerTest (5)
OnnxTransformTests.cs (5)
910new[] { nameof(PredictionWithCustomShape.argmax) }, new[] { nameof(InputWithCustomShape.input) }, 917nameof(PredictionWithCustomShape.argmax), nameof(InputWithCustomShape.input), 930var transformedDataPoints = ML.Data.CreateEnumerable<PredictionWithCustomShape>(transformedDataView, false).ToList(); 1072var pipeline = ML.Transforms.ApplyOnnxModel(nameof(PredictionWithCustomShape.argmax), 1094var transformedDataPoints = ML.Data.CreateEnumerable<PredictionWithCustomShape>(transformedDataView, false).ToList();