1 write to Options
Microsoft.ML.TorchSharp (1)
TorchSharpBaseTrainer.cs (1)
393Options = options;
15 references to Options
Microsoft.ML.TorchSharp (15)
NasBert\NasBertTrainer.cs (9)
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()); 536info[1] = new DataViewSchema.DetachedColumn(Parent.Options.ScoreColumnName, new VectorDataViewType(NumberDataViewType.Single, Parent.Options.NumberOfClasses), meta.ToAnnotations()); 548info[0] = new DataViewSchema.DetachedColumn(Parent.Options.PredictionColumnName, new VectorDataViewType(new KeyDataViewType(typeof(uint), Parent.Options.NumberOfClasses - 1)), labelBuilder.ToAnnotations()); 556info[0] = new DataViewSchema.DetachedColumn(Parent.Options.ScoreColumnName, NumberDataViewType.Single);
NasBert\TextClassificationTrainer.cs (1)
332var editor = VBufferEditor.Create(ref dst, Parent.Options.NumberOfClasses);
TorchSharpBaseTrainer.cs (5)
395ScoreColumnName = Options.ScoreColumnName; 428ctx.SaveNonEmptyString(Options.LabelColumnName); 429ctx.SaveStringOrNull(Options.ScoreColumnName); 430ctx.SaveNonEmptyString(Options.PredictionColumnName); 431ctx.Writer.Write(Options.NumberOfClasses);