using TextLoader.Create instead of env.CreateLoader (#1025)

dotnet · Sep 25, 2018 · 330aa41 · 330aa41
1 parent 893a385
commit 330aa41
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/test/Microsoft.ML.Tests/ScenariosWithDirectInstantiation/TensorflowTests.cs b/test/Microsoft.ML.Tests/ScenariosWithDirectInstantiation/TensorflowTests.cs
@@ -563,7 +563,14 @@ public void TensorFlowTransformCifarSavedModel()
                 var imageWidth = 32;
                 var dataFile = GetDataPath("images/images.tsv");
                 var imageFolder = Path.GetDirectoryName(dataFile);
-                var data = env.CreateLoader("Text{col=ImagePath:TX:0 col=Name:TX:1}", new MultiFileSource(dataFile));
+                var data = TextLoader.Create(env, new TextLoader.Arguments()
+                {
+                    Column = new[]
+                    {
+                        new TextLoader.Column("ImagePath", DataKind.TX, 0),
+                        new TextLoader.Column("Name", DataKind.TX, 1),
+                    }
+                }, new MultiFileSource(dataFile));
                 var images = ImageLoaderTransform.Create(env, new ImageLoaderTransform.Arguments()
                 {
                     Column = new ImageLoaderTransform.Column[1]