3 references to ModelForPrediction
Microsoft.ML.TorchSharp (3)
NasBert\NasBertTrainer.cs (1)
210
model = new
ModelForPrediction
(Parent.BertOptions, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, Parent.Option.NumberOfClasses);
NasBert\SentenceSimilarityTrainer.cs (1)
244
var model = new
ModelForPrediction
(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.NumberOfClasses);
NasBert\TextClassificationTrainer.cs (1)
268
var model = new
ModelForPrediction
(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.NumberOfClasses);