10 writes to TaskType
Microsoft.ML.TorchSharp (9)
NasBert\NerTrainer.cs (2)
75
TaskType
= BertTaskType.NamedEntityRecognition;
338
options.
TaskType
= BertTaskType.NamedEntityRecognition;
NasBert\SentenceSimilarityTrainer.cs (3)
68
TaskType
= BertTaskType.SentenceRegression;
95
TaskType
= BertTaskType.SentenceRegression,
250
options.
TaskType
= BertTaskType.SentenceRegression;
NasBert\TextClassificationTrainer.cs (3)
67
TaskType
= BertTaskType.TextClassification;
97
TaskType
= BertTaskType.TextClassification
274
options.
TaskType
= BertTaskType.TextClassification;
Roberta\QATrainer.cs (1)
81
TaskType
= BertTaskType.QuestionAnswering;
Microsoft.ML.TorchSharp.Tests (1)
TextClassificationTests.cs (1)
436
TaskType
= BertTaskType.SentenceRegression,
15 references to TaskType
Microsoft.ML.TorchSharp (15)
NasBert\Models\BaseModel.cs (1)
15
public BertTaskType HeadType => Options.
TaskType
;
NasBert\NasBertTrainer.cs (14)
173
Contracts.Assert(BertOptions.
TaskType
!= BertTaskType.None, "BertTaskType must be specified");
207
if (Parent.BertOptions.
TaskType
== BertTaskType.NamedEntityRecognition)
271
if (Parent.BertOptions.
TaskType
== BertTaskType.NamedEntityRecognition)
294
if (Parent.BertOptions.
TaskType
== BertTaskType.TextClassification)
296
else if (Parent.BertOptions.
TaskType
== BertTaskType.NamedEntityRecognition)
325
if (BertOptions.
TaskType
== BertTaskType.TextClassification)
341
else if (BertOptions.
TaskType
== BertTaskType.NamedEntityRecognition)
374
if (BertOptions.
TaskType
== BertTaskType.TextClassification)
390
else if (BertOptions.
TaskType
== BertTaskType.NamedEntityRecognition)
434
if (BertOptions.
TaskType
== BertTaskType.TextClassification)
464
if (BertOptions.Sentence2ColumnName != default || BertOptions.
TaskType
== BertTaskType.SentenceRegression)
510
if (Parent.BertOptions.Sentence2ColumnName != default || Parent.BertOptions.
TaskType
== BertTaskType.SentenceRegression)
517
if (Parent.BertOptions.
TaskType
== BertTaskType.TextClassification)
539
else if (Parent.BertOptions.
TaskType
== BertTaskType.NamedEntityRecognition)