4 types derived from NasBertOptions
Microsoft.ML.TorchSharp (4)
NasBert\NerTrainer.cs (1)
67public class NerOptions : NasBertOptions
NasBert\SentenceSimilarityTrainer.cs (1)
62public class SentenceSimilarityOptions : NasBertOptions
NasBert\TextClassificationTrainer.cs (1)
63public class TextClassificationOptions : NasBertTrainer.NasBertOptions
Roberta\QATrainer.cs (1)
38public sealed class Options : NasBertOptions
2 instantiations of NasBertOptions
Microsoft.ML.TorchSharp (2)
NasBert\SentenceSimilarityTrainer.cs (1)
232var options = new NasBertOptions()
NasBert\TextClassificationTrainer.cs (1)
256var options = new NasBertOptions()
21 references to NasBertOptions
Microsoft.ML.TorchSharp (21)
NasBert\Models\BaseModel.cs (2)
12protected readonly NasBertTrainer.NasBertOptions Options; 24protected BaseModel(NasBertTrainer.NasBertOptions options)
NasBert\Models\ModelPrediction.cs (1)
19public ModelForPrediction(NasBertTrainer.NasBertOptions options, int padIndex, int symbolsCount, int numClasses)
NasBert\Models\NasBertModel.cs (1)
19public NasBertModel(NasBertTrainer.NasBertOptions options, int padIndex, int symbolsCount)
NasBert\Models\NerModel.cs (1)
18public NerModel(NasBertTrainer.NasBertOptions options, int padIndex, int symbolsCount, int numLabels)
NasBert\NasBertTrainer.cs (4)
167internal readonly NasBertTrainer.NasBertOptions BertOptions; 171BertOptions = options as NasBertTrainer.NasBertOptions; 413internal readonly NasBertTrainer.NasBertOptions BertOptions; 419internal NasBertTransformer(IHostEnvironment env, NasBertTrainer.NasBertOptions options, NasBertModel model, DataViewSchema.DetachedColumn labelColumn)
NasBert\NerTrainer.cs (2)
111return new NerTransformer(host, options as NasBertOptions, model as NasBertModel, labelColumn); 258internal NerTransformer(IHostEnvironment env, NasBertOptions options, NasBertModel model, DataViewSchema.DetachedColumn labelColumn) : base(env, options, model, labelColumn)
NasBert\Optimizers\Adam.cs (1)
15public Adam(NasBertTrainer.NasBertOptions options, IEnumerable<Parameter> parameters)
NasBert\Optimizers\BaseOptimizer.cs (3)
26public static BaseOptimizer GetOptimizer(NasBertTrainer.NasBertOptions options, IEnumerable<Parameter> parameters) 31protected NasBertTrainer.NasBertOptions Options { get; set; } 37protected BaseOptimizer(string name, NasBertTrainer.NasBertOptions options, IEnumerable<Parameter> parameters)
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) 232var options = new NasBertOptions()
NasBert\TextClassificationTrainer.cs (3)
109return new TextClassificationTransformer(host, options as NasBertOptions, model as NasBertModel, labelColumn); 194internal TextClassificationTransformer(IHostEnvironment env, NasBertOptions options, NasBertModel model, DataViewSchema.DetachedColumn labelColumn) : base(env, options, model, labelColumn) 256var options = new NasBertOptions()