Move options

nicholasmerci · Oct 6, 2020 · dc924d4 · dc924d4
1 parent 6b7dbb6
commit dc924d4
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/MLNetExamples/DeepNeuralNetworkUpdate/Program.cs b/MLNetExamples/DeepNeuralNetworkUpdate/Program.cs
@@ -33,6 +33,13 @@ static void Main(string[] args)
                 .Fit(testTrainData.TestSet)
                 .Transform(testTrainData.TestSet);
 
+            var imagesPipeline = context.Transforms.Conversion.MapValueToKey("LabelKey", "Label", keyOrdinality: Microsoft.ML.Transforms.ValueToKeyMappingEstimator.KeyOrdinality.ByValue)
+                .Append(context.Transforms.LoadRawImageBytes("Image", imagesFolder, "ImagePath"));
+
+            var imageDataModel = imagesPipeline.Fit(testTrainData.TrainSet);
+
+            var imageDataView = imageDataModel.Transform(testTrainData.TrainSet);
+
             var options = new ImageClassificationTrainer.Options()
             {
                 Arch = ImageClassificationTrainer.Architecture.ResnetV250,
@@ -44,13 +51,6 @@ static void Main(string[] args)
                 ValidationSet = validationData
             };
 
-            var imagesPipeline = context.Transforms.Conversion.MapValueToKey("LabelKey", "Label", keyOrdinality: Microsoft.ML.Transforms.ValueToKeyMappingEstimator.KeyOrdinality.ByValue)
-                .Append(context.Transforms.LoadRawImageBytes("Image", imagesFolder, "ImagePath"));
-
-            var imageDataModel = imagesPipeline.Fit(testTrainData.TrainSet);
-
-            var imageDataView = imageDataModel.Transform(testTrainData.TrainSet);
-
             var pipeline = context.MulticlassClassification.Trainers.ImageClassification(options)
                 .Append(context.Transforms.Conversion.MapKeyToValue("PredictedLabel"));