6 writes to NumberOfClasses
Microsoft.ML.TorchSharp (6)
NasBert\NerTrainer.cs (2)
225
Parent.Option.
NumberOfClasses
= keys.Length + 1;
325
NumberOfClasses
= ctx.Reader.ReadInt32(),
NasBert\SentenceSimilarityTrainer.cs (2)
157
Parent.Option.
NumberOfClasses
= 1;
237
NumberOfClasses
= ctx.Reader.ReadInt32(),
NasBert\TextClassificationTrainer.cs (2)
160
Parent.Option.
NumberOfClasses
= uniqueLabels.Count;
261
NumberOfClasses
= ctx.Reader.ReadInt32(),
10 references to NumberOfClasses
Microsoft.ML.TorchSharp (10)
NasBert\NasBertTrainer.cs (5)
208
model = new NerModel(Parent.BertOptions, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, Parent.Option.
NumberOfClasses
);
210
model = new ModelForPrediction(Parent.BertOptions, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, Parent.Option.
NumberOfClasses
);
534
info[0] = new DataViewSchema.DetachedColumn(Parent.Options.PredictionColumnName, new KeyDataViewType(typeof(uint), Parent.Options.
NumberOfClasses
), labelBuilder.ToAnnotations());
536
info[1] = new DataViewSchema.DetachedColumn(Parent.Options.ScoreColumnName, new VectorDataViewType(NumberDataViewType.Single, Parent.Options.
NumberOfClasses
), meta.ToAnnotations());
548
info[0] = new DataViewSchema.DetachedColumn(Parent.Options.PredictionColumnName, new VectorDataViewType(new KeyDataViewType(typeof(uint), Parent.Options.
NumberOfClasses
- 1)), labelBuilder.ToAnnotations());
NasBert\NerTrainer.cs (1)
332
var model = new NerModel(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.
NumberOfClasses
);
NasBert\SentenceSimilarityTrainer.cs (1)
244
var model = new ModelForPrediction(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.
NumberOfClasses
);
NasBert\TextClassificationTrainer.cs (2)
268
var model = new ModelForPrediction(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.
NumberOfClasses
);
332
var editor = VBufferEditor.Create(ref dst, Parent.Options.
NumberOfClasses
);
TorchSharpBaseTrainer.cs (1)
431
ctx.Writer.Write(Options.
NumberOfClasses
);