3 instantiations of ModelForPrediction
Microsoft.ML.TorchSharp (3)
NasBert\NasBertTrainer.cs (1)
210model = new ModelForPrediction(Parent.BertOptions, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, Parent.Option.NumberOfClasses);
NasBert\SentenceSimilarityTrainer.cs (1)
244var model = new ModelForPrediction(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.NumberOfClasses);
NasBert\TextClassificationTrainer.cs (1)
268var model = new ModelForPrediction(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.NumberOfClasses);
4 references to ModelForPrediction
Microsoft.ML.TorchSharp (4)
NasBert\SentenceSimilarityTrainer.cs (3)
109return new SentenceSimilarityTransformer(host, options as NasBertOptions, model as ModelForPrediction, labelColumn); 189internal SentenceSimilarityTransformer(IHostEnvironment env, NasBertOptions options, ModelForPrediction model, DataViewSchema.DetachedColumn labelColumn) : base(env, options, model, labelColumn) 244var model = new ModelForPrediction(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.NumberOfClasses);
NasBert\TextClassificationTrainer.cs (1)
268var model = new ModelForPrediction(options, tokenizerModel.PadIndex, tokenizerModel.SymbolsCount, options.NumberOfClasses);