27 references to BertTaskType
Microsoft.ML.TorchSharp (26)
NasBert\Models\BaseModel.cs (1)
15
public
BertTaskType
HeadType => Options.TaskType;
NasBert\NasBertTrainer.cs (16)
118
public
BertTaskType
TaskType =
BertTaskType
.None;
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)
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,