diff --git a/src/Microsoft.ML.TensorFlow/doc.xml b/src/Microsoft.ML.TensorFlow/doc.xml index 68a4d4ddeb..6f89d24f0f 100644 --- a/src/Microsoft.ML.TensorFlow/doc.xml +++ b/src/Microsoft.ML.TensorFlow/doc.xml @@ -8,34 +8,34 @@ - The TensorflowTransform extracts the specified outputs from the operations computed on the graph (given the input(s)) using a pre-trained Tensorflow model. - The transform takes as input the Tensorflow model together with the names of the inputs to the model and names of the operations for which output values will be extracted from the model. + The TensorFlowTransform extracts specified outputs using a pre-trained Tensorflow model. + The transform takes as inputs the pre-trained Tensorflow model, the names of the input nodes, and names of the output nodes whose values we want to extract. - This transform requires the Microsoft.ML.TensorFlow nuget to be installed. - The TensorflowTransform has following assumptions regarding the input, output and processing of data. + This transform requires the Microsoft.ML.TensorFlow nuget to be installed. + The TensorFlowTransform has the following assumptions regarding input, output and processing of data. - The transform currently accepts the frozen TensorFlow model file as input. + For the input model, currently the TensorFlowTransform supports both the Frozen model format and also the SavedModel format. The transform supports scoring only one example at a time. - The name of input column(s) should match the name of input(s) in Tensorflow model. + The name of input column(s) should match the name of input(s) in TensorFlow model. - The name of each output column should match one of the operations in the Tensorflow graph. + The name of each output column should match one of the operations in the TensorFlow graph. Currently, float and double are the only acceptable data types for input/output. - Upon success, the transform will introduce a new column in corresponding to each output column specified. + Upon success, the transform will introduce a new column in corresponding to each output column specified. @@ -50,16 +50,16 @@ pipeline.Add(new ColumnCopier(("NumericImageVec", "Input"); pipeline.Add(new TensorFlowScorer() { - ModelFile = model_location; - InputColumns = new []{ "Input" }; - OutputColumns = "Output" + Model = model_location; + InputColumns = new[] { "Input" }; + OutputColumns = new[] { "Output" }; } - var pipeline = new LearningPipeline(seed: 1); - pipeline.Add(new TextLoader(dataFile).CreateFrom<CifarData>(useHeader: false)); + var pipeline = new Legacy.LearningPipeline(seed: 1); + pipeline.Add(new Microsoft.ML.Legacy.Data.TextLoader(dataFile).CreateFrom<CifarData>(useHeader: false)); pipeline.Add(new ImageLoader(("ImagePath", "ImageReal")) { ImageFolder = imageFolder @@ -80,10 +80,11 @@ pipeline.Add(new TensorFlowScorer() { - ModelFile = model_location, + Model = model_location, InputColumns = new[] { "Input" }, - OutputColumns = "Output" + OutputColumns = new[] { "Output" } }); +