Skip to content

Commit 330aa41

Browse files
abgoswamIvanidzo4ka
authored andcommitted
using TextLoader.Create instead of env.CreateLoader (dotnet#1025)
1 parent 893a385 commit 330aa41

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

test/Microsoft.ML.Tests/ScenariosWithDirectInstantiation/TensorflowTests.cs

+8-1
Original file line numberDiff line numberDiff line change
@@ -563,7 +563,14 @@ public void TensorFlowTransformCifarSavedModel()
563563
var imageWidth = 32;
564564
var dataFile = GetDataPath("images/images.tsv");
565565
var imageFolder = Path.GetDirectoryName(dataFile);
566-
var data = env.CreateLoader("Text{col=ImagePath:TX:0 col=Name:TX:1}", new MultiFileSource(dataFile));
566+
var data = TextLoader.Create(env, new TextLoader.Arguments()
567+
{
568+
Column = new[]
569+
{
570+
new TextLoader.Column("ImagePath", DataKind.TX, 0),
571+
new TextLoader.Column("Name", DataKind.TX, 1),
572+
}
573+
}, new MultiFileSource(dataFile));
567574
var images = ImageLoaderTransform.Create(env, new ImageLoaderTransform.Arguments()
568575
{
569576
Column = new ImageLoaderTransform.Column[1]

0 commit comments

Comments
 (0)