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