1 write to Option
Microsoft.ML.TorchSharp (1)
TorchSharpBaseTrainer.cs (1)
94Option = options;
44 references to Option
Microsoft.ML.TorchSharp (44)
NasBert\NasBertTrainer.cs (24)
194total_steps: ((TrainingRowCount / Parent.Option.BatchSize) + 1) * Parent.Option.MaxEpoch, 197div_factor: 1.0 / Parent.Option.StartLearningRateRatio, 198final_div_factor: Parent.Option.StartLearningRateRatio / Parent.Option.FinalLearningRateRatio); 208model = new NerModel(Parent.BertOptions, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, Parent.Option.NumberOfClasses); 210model = new ModelForPrediction(Parent.BertOptions, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, Parent.Option.NumberOfClasses); 219return input.GetRowCursor(input.Schema[Parent.BertOptions.Sentence1ColumnName], input.Schema[Parent.BertOptions.Sentence2ColumnName], input.Schema[Parent.Option.LabelColumnName]); 221return input.GetRowCursor(input.Schema[Parent.BertOptions.Sentence1ColumnName], input.Schema[Parent.Option.LabelColumnName]); 333inputSchema.TryFindColumn(Option.LabelColumnName, out var labelCol); 335outColumns[Option.PredictionColumnName] = new SchemaShape.Column(Option.PredictionColumnName, SchemaShape.Column.VectorKind.Scalar, 338outColumns[Option.ScoreColumnName] = new SchemaShape.Column(Option.ScoreColumnName, SchemaShape.Column.VectorKind.Vector, 348inputSchema.TryFindColumn(Option.LabelColumnName, out var labelCol); 350outColumns[Option.PredictionColumnName] = new SchemaShape.Column(Option.PredictionColumnName, SchemaShape.Column.VectorKind.VariableVector, 355outColumns[Option.ScoreColumnName] = new SchemaShape.Column(Option.ScoreColumnName, SchemaShape.Column.VectorKind.Scalar, 371if (!inputSchema.TryFindColumn(Option.LabelColumnName, out var labelCol)) 372throw Host.ExceptSchemaMismatch(nameof(inputSchema), "label", Option.LabelColumnName); 377throw Host.ExceptSchemaMismatch(nameof(inputSchema), "label", Option.LabelColumnName, 393throw Host.ExceptSchemaMismatch(nameof(inputSchema), "label", Option.LabelColumnName, 399throw Host.ExceptSchemaMismatch(nameof(inputSchema), "label", Option.LabelColumnName,
NasBert\NerTrainer.cs (3)
216input.Schema[Parent.Option.LabelColumnName].GetKeyValues(ref keys); 217var labelCol = input.GetColumn<VBuffer<uint>>(Parent.Option.LabelColumnName); 225Parent.Option.NumberOfClasses = keys.Length + 1;
NasBert\SentenceSimilarityTrainer.cs (2)
148var labelCol = input.GetColumn<float>(Parent.Option.LabelColumnName); 157Parent.Option.NumberOfClasses = 1;
NasBert\TextClassificationTrainer.cs (2)
150var labelCol = input.GetColumn<uint>(Parent.Option.LabelColumnName); 160Parent.Option.NumberOfClasses = uniqueLabels.Count;
TorchSharpBaseTrainer.cs (13)
109for (int i = 0; i < Option.MaxEpoch; i++) 115if (Option.ValidationSet != null) 118var labelCol = input.Schema.GetColumnOrNull(Option.LabelColumnName); 120transformer = CreateTransformer(Host, Option, trainer.Model, new DataViewSchema.DetachedColumn(labelCol.Value)); 197var validationSet = Parent.Option.ValidationSet; 203var labelGetter = cursor.GetGetter<TLabelCol>(validationSet.Schema[Parent.Option.LabelColumnName]); 206List<Tensor> inputTensors = new List<Tensor>(Parent.Option.BatchSize); 207List<TTargetsCol> targets = new List<TTargetsCol>(Parent.Option.BatchSize); 237for (int i = 0; i < Parent.Option.BatchSize && cursorValid; i++) 275var labelGetter = cursor.GetGetter<TLabelCol>(input.Schema[Parent.Option.LabelColumnName]); 278List<Tensor> inputTensors = new List<Tensor>(Parent.Option.BatchSize); 279List<TTargetsCol> targets = new List<TTargetsCol>(Parent.Option.BatchSize); 310for (int i = 0; i < Parent.Option.BatchSize && cursorValid; i++)