2 instantiations of QATrainer
Microsoft.ML.TorchSharp (2)
TorchSharpCatalog.cs (2)
270=> new QATrainer(CatalogUtils.GetEnvironment(catalog), contextColumnName, questionColumnName, trainingAnswerColumnName, answerIndexColumnName, predictedAnswerColumnName, scoreColumnName, topK, batchSize, maxEpochs, validationSet, architecture); 283=> new QATrainer(CatalogUtils.GetEnvironment(catalog), options);
12 references to QATrainer
Microsoft.ML.TorchSharp (11)
Roberta\Models\RobertaModel.cs (1)
39protected RobertaModel(QATrainer.Options options)
Roberta\Models\RobertaModelForQA.cs (1)
21public RobertaModelForQA(QATrainer.Options options)
Roberta\QATrainer.cs (6)
104Host = Contracts.CheckRef(env, nameof(env)).Register(nameof(QATrainer)); 185protected readonly QATrainer Parent; 192public Trainer(QATrainer parent, IChannel ch, IDataView input) 561internal readonly QATrainer.Options Options; 572internal QATransformer(IHostEnvironment env, QATrainer.Options options, RobertaModelForQA model) 688var options = new QATrainer.Options()
TorchSharpCatalog.cs (3)
257public static QATrainer QuestionAnswer( 280public static QATrainer QuestionAnswer( 282QATrainer.Options options)
Microsoft.ML.TorchSharp.Tests (1)
QATests.cs (1)
87var estimator = ML.MulticlassClassification.Trainers.QuestionAnswer(maxEpochs: 30);