7 writes to PredictionColumnName
Microsoft.ML.TorchSharp (5)
NasBert\NerTrainer.cs (2)
92PredictionColumnName = predictionColumnName, 324PredictionColumnName = ctx.LoadString(),
NasBert\SentenceSimilarityTrainer.cs (1)
237PredictionColumnName = ctx.LoadString(),
NasBert\TextClassificationTrainer.cs (2)
90PredictionColumnName = predictionColumnName, 261PredictionColumnName = ctx.LoadString(),
Microsoft.ML.TorchSharp.Tests (2)
NerTests.cs (2)
123options.PredictionColumnName = "outputColumn"; 220options.PredictionColumnName = "outputColumn";
10 references to PredictionColumnName
Microsoft.ML.TorchSharp (10)
NasBert\NasBertTrainer.cs (8)
335outColumns[Option.PredictionColumnName] = new SchemaShape.Column(Option.PredictionColumnName, SchemaShape.Column.VectorKind.Scalar, 350outColumns[Option.PredictionColumnName] = new SchemaShape.Column(Option.PredictionColumnName, SchemaShape.Column.VectorKind.VariableVector, 440outColumns[Options.PredictionColumnName] = new SchemaShape.Column(Options.PredictionColumnName, SchemaShape.Column.VectorKind.Scalar, 534info[0] = new DataViewSchema.DetachedColumn(Parent.Options.PredictionColumnName, new KeyDataViewType(typeof(uint), Parent.Options.NumberOfClasses), labelBuilder.ToAnnotations()); 548info[0] = new DataViewSchema.DetachedColumn(Parent.Options.PredictionColumnName, new VectorDataViewType(new KeyDataViewType(typeof(uint), Parent.Options.NumberOfClasses - 1)), labelBuilder.ToAnnotations());
TorchSharpBaseTrainer.cs (2)
93Contracts.AssertValue(options.PredictionColumnName); 430ctx.SaveNonEmptyString(Options.PredictionColumnName);