4 instantiations of OnnxTransformer
Microsoft.ML.OnnxTransformer (4)
OnnxTransform.cs (4)
149
return new
OnnxTransformer
(env, options).MakeDataTransform(input);
225
return new
OnnxTransformer
(env, options, modelBytes);
905
: this(env, new
OnnxTransformer
(env, new string[] { }, new string[] { }, modelFile, gpuDeviceId, fallbackToCpu, shapeDictionary, recursionLimit))
926
: this(env, new
OnnxTransformer
(env, outputColumnNames, inputColumnNames, modelFile, gpuDeviceId, fallbackToCpu, shapeDictionary, recursionLimit, interOpNumThreads, intraOpNumThreads))
51 references to OnnxTransformer
Microsoft.ML.IntegrationTests (3)
ONNX.cs (3)
61
var
onnxModel = onnxEstimator.Fit(data);
107
var
onnxModel = onnxEstimator.Fit(data);
159
var
onnxModel = onnxEstimator.Fit(data);
Microsoft.ML.OnnxTransformer (26)
OnnxTransform.cs (24)
21
[assembly: LoadableClass(
OnnxTransformer
.Summary, typeof(IDataTransform), typeof(
OnnxTransformer
),
22
typeof(
OnnxTransformer
.Options), typeof(SignatureDataTransform),
OnnxTransformer
.UserName,
OnnxTransformer
.ShortName, "OnnxTransform", "OnnxScorer")]
24
[assembly: LoadableClass(
OnnxTransformer
.Summary, typeof(IDataTransform), typeof(
OnnxTransformer
),
25
null, typeof(SignatureLoadDataTransform),
OnnxTransformer
.UserName,
OnnxTransformer
.LoaderSignature)]
27
[assembly: LoadableClass(typeof(
OnnxTransformer
), null, typeof(SignatureLoadModel),
28
OnnxTransformer
.UserName,
OnnxTransformer
.LoaderSignature)]
30
[assembly: LoadableClass(typeof(IRowMapper), typeof(
OnnxTransformer
), null, typeof(SignatureLoadRowMapper),
31
OnnxTransformer
.UserName,
OnnxTransformer
.LoaderSignature)]
33
[assembly: EntryPointModule(typeof(
OnnxTransformer
))]
143
loaderAssemblyName: typeof(
OnnxTransformer
).Assembly.FullName);
157
private static
OnnxTransformer
Create(IHostEnvironment env, ModelLoadContext ctx)
233
base(Contracts.CheckRef(env, nameof(env)).Register(nameof(
OnnxTransformer
)))
420
private readonly
OnnxTransformer
_parent;
435
public Mapper(
OnnxTransformer
parent, DataViewSchema inputSchema) :
889
public sealed class OnnxScoringEstimator : TrivialEstimator<
OnnxTransformer
>
930
internal OnnxScoringEstimator(IHostEnvironment env,
OnnxTransformer
transformer)
931
: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(
OnnxTransformer
)), transformer)
OnnxUtils.cs (2)
151
/// For example, ML.NET can shuffle the inputs so that the i-th ONNX input becomes the j-th input column of <see cref="
OnnxTransformer
"/>.
152
/// ML.NET can also only exposes a subset of ONNX outputs in <see cref="
OnnxTransformer
"/>.
Microsoft.ML.OnnxTransformerTest (11)
OnnxTransformTests.cs (11)
168
var
onnxTransformer = pipe.Fit(invalidDataWrongVectorSize);
196
var
transformer = est.Fit(dataView);
316
var
onnxTransformer = pipeline.Fit(dataView);
349
var
onnxTransformer = pipeline.Fit(dataView);
389
var
onnxTransformer = pipeline.Fit(dataView);
449
var
onnxTransformer = pipeline.Fit(idv);
475
var
onnxTransformer = pipeline.Fit(idv);
607
var
onnxTransformer = pipeline.Fit(dataView);
660
var
onnxTransformer = pipeline.Fit(dataView);
815
var onnxTransformer = new
OnnxTransformer
[3];
987
var
model = pipeline.Fit(dataView);
Microsoft.ML.Samples (1)
Dynamic\ModelOperations\OnnxConversion.cs (1)
86
using
var
onnxTransformer = onnxEstimator.Fit(trainTestOriginalData.TrainSet);
Microsoft.ML.Tests (10)
OnnxConversionTest.cs (8)
827
var
onnxTransformer = onnxEstimator.Fit(data);
989
var
onnxTransformer = onnxEstimator.Fit(dataView);
1516
var
onnxTransformer = onnxEstimator.Fit(dataView);
1633
var
onnxTransformer = onnxEstimator.Fit(dataView);
1744
var
loadedModel = pipeline.Fit(dataView);
1807
var
onnxTransformer = onnxEstimator.Fit(reloadedData);
1820
var
onnxTransformer2 = onnxEstimator2.Fit(originalData);
2255
var
onnxTransformer = onnxEstimator.Fit(dataView);
OnnxSequenceTypeWithAttributesTest.cs (2)
52
var
model = pipeline.Fit(dataView);
91
var
model = pipeline.Fit(dataView);