2 instantiations of QATransformer
Microsoft.ML.TorchSharp (2)
Roberta\QATrainer.cs (2)
173transformer = new QATransformer(Host, Option, trainer.Model); 704return new QATransformer(env, options, model);
15 references to QATransformer
Microsoft.ML.TorchSharp (14)
Roberta\QATrainer.cs (14)
28[assembly: LoadableClass(typeof(QATransformer), null, typeof(SignatureLoadModel), 29QATransformer.UserName, QATransformer.LoaderSignature)] 31[assembly: LoadableClass(typeof(IRowMapper), typeof(QATransformer), null, typeof(SignatureLoadRowMapper), 32QATransformer.UserName, QATransformer.LoaderSignature)] 36public class QATrainer : IEstimator<QATransformer> 144public QATransformer Fit(IDataView input) 148QATransformer transformer = default; 573: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(QATransformer))) 638loaderAssemblyName: typeof(QATransformer).Assembly.FullName); 674private static QATransformer Create(IHostEnvironment env, ModelLoadContext ctx) 709private readonly QATransformer _parent; 712public QAMapper(QATransformer parent, DataViewSchema inputSchema) :
Microsoft.ML.TorchSharp.Tests (1)
QATests.cs (1)
88var model = estimator.Fit(dataView);