4 types derived from NasBertOptions
Microsoft.ML.TorchSharp (4)
NasBert\NerTrainer.cs (1)
67public class NerOptions : NasBertOptions
NasBert\SentenceSimilarityTrainer.cs (1)
63public class SentenceSimilarityOptions : NasBertOptions
NasBert\TextClassificationTrainer.cs (1)
64public 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)
233var options = new NasBertOptions()
NasBert\TextClassificationTrainer.cs (1)
257var 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)
110return new SentenceSimilarityTransformer(host, options as NasBertOptions, model as ModelForPrediction, labelColumn); 190internal SentenceSimilarityTransformer(IHostEnvironment env, NasBertOptions options, ModelForPrediction model, DataViewSchema.DetachedColumn labelColumn) : base(env, options, model, labelColumn) 233var options = new NasBertOptions()
NasBert\TextClassificationTrainer.cs (3)
110return new TextClassificationTransformer(host, options as NasBertOptions, model as NasBertModel, labelColumn); 195internal TextClassificationTransformer(IHostEnvironment env, NasBertOptions options, NasBertModel model, DataViewSchema.DetachedColumn labelColumn) : base(env, options, model, labelColumn) 257var options = new NasBertOptions()