4 instantiations of TensorFlowTransformer
Microsoft.ML.TensorFlow (4)
TensorflowTransform.cs (4)
151
return new
TensorFlowTransformer
(env, LoadTFSession(env, modelBytes), outputs, inputs, null, false, addBatchDimensionInput, treatOutputAsBatched: treatOutputAsBatched);
180
return new
TensorFlowTransformer
(env, GetSession(env, tempDirPath), outputs, inputs, tempDirPath, true, addBatchDimensionInput, treatOutputAsBatched: treatOutputAsBatched);
198
return new
TensorFlowTransformer
(env, options, input).MakeDataTransform(input);
1061
_transformer = new
TensorFlowTransformer
(_host, _tensorFlowModel.Session, _options.OutputColumns, _options.InputColumns,
41 references to TensorFlowTransformer
Microsoft.ML.Core.Tests (3)
UnitTests\TestEntryPoints.cs (3)
328
Env.ComponentCatalog.RegisterAssembly(typeof(
TensorFlowTransformer
).Assembly);
6475
Env.ComponentCatalog.RegisterAssembly(typeof(
TensorFlowTransformer
).Assembly);
6491
Env.ComponentCatalog.RegisterAssembly(typeof(
TensorFlowTransformer
).Assembly);
Microsoft.ML.Samples (1)
Dynamic\TensorFlow\ImageClassification.cs (1)
48
var
estimator = pipeline.Fit(idv);
Microsoft.ML.Samples.GPU (1)
docs\samples\Microsoft.ML.Samples\Dynamic\TensorFlow\ImageClassification.cs (1)
48
var
estimator = pipeline.Fit(idv);
Microsoft.ML.TensorFlow (29)
TensorflowTransform.cs (28)
25
[assembly: LoadableClass(
TensorFlowTransformer
.Summary, typeof(IDataTransform), typeof(
TensorFlowTransformer
),
26
typeof(TensorFlowEstimator.Options), typeof(SignatureDataTransform),
TensorFlowTransformer
.UserName,
TensorFlowTransformer
.ShortName)]
28
[assembly: LoadableClass(
TensorFlowTransformer
.Summary, typeof(IDataTransform), typeof(
TensorFlowTransformer
), null, typeof(SignatureLoadDataTransform),
29
TensorFlowTransformer
.UserName,
TensorFlowTransformer
.LoaderSignature)]
31
[assembly: LoadableClass(typeof(
TensorFlowTransformer
), null, typeof(SignatureLoadModel),
32
TensorFlowTransformer
.UserName,
TensorFlowTransformer
.LoaderSignature)]
34
[assembly: LoadableClass(typeof(IRowMapper), typeof(
TensorFlowTransformer
), null, typeof(SignatureLoadRowMapper),
35
TensorFlowTransformer
.UserName,
TensorFlowTransformer
.LoaderSignature)]
37
[assembly: EntryPointModule(typeof(
TensorFlowTransformer
))]
88
loaderAssemblyName: typeof(
TensorFlowTransformer
).Assembly.FullName);
128
private static
TensorFlowTransformer
Create(IHostEnvironment env, ModelLoadContext ctx)
154
var tempDirPath = Path.GetFullPath(Path.Combine(((IHostEnvironmentInternal)env).TempFilePath, nameof(
TensorFlowTransformer
) + "_" + Guid.NewGuid()));
289
: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(
TensorFlowTransformer
)))
536
private readonly
TensorFlowTransformer
_parent;
542
public Mapper(
TensorFlowTransformer
parent, DataViewSchema inputSchema) :
927
public sealed class TensorFlowEstimator : IEstimator<
TensorFlowTransformer
>
930
/// The options for the <see cref="
TensorFlowTransformer
"/>.
983
private
TensorFlowTransformer
_transformer;
1009
var inputTuple =
TensorFlowTransformer
.GetInputInfo(_host, tensorFlowModel.Session, _options.InputColumns);
1011
var outputTuple =
TensorFlowTransformer
.GetOutputInfo(_host, tensorFlowModel.Session, _options.OutputColumns, _options.TreatOutputAsBatched);
1054
/// Trains and returns a <see cref="
TensorFlowTransformer
"/>.
1056
public
TensorFlowTransformer
Fit(IDataView input)
TensorflowUtils.cs (1)
121
/// <see cref="
TensorFlowTransformer
"/>, the output schema contains a column with the name of that node, and the
Microsoft.ML.TensorFlow.Tests (7)
TensorFlowEstimatorTests.cs (2)
125
var
transformer = est.Fit(dataView);
141
Type type = typeof(
TensorFlowTransformer
);
TensorflowTests.cs (5)
200
var
trans = tfModel.ScoreTensorFlowModel(new[] { "c" }, new[] { "a", "b" }).Fit(loader);
292
var
trans = tfModel.ScoreTensorFlowModel(outputs, inputs).Fit(loader);
414
var
trans = tfModel.ScoreTensorFlowModel(outputs, inputs).Fit(loader);
1027
var
trans = tensorFlowModel.ScoreTensorFlowModel("Output", "Input").Fit(pixels);
1079
var
trans = tensorFlowModel.ScoreTensorFlowModel("Output", "Input").Fit(pixels);