156 references to TaskKind
Microsoft.ML.AutoML (53)
API\BinaryClassificationExperiment.cs (1)
152
TaskKind
.BinaryClassification,
API\ExperimentBase.cs (6)
28
private readonly
TaskKind
_task;
35
TaskKind
task,
71
if (_task ==
TaskKind
.Ranking)
135
if (_task ==
TaskKind
.Ranking)
161
var columnInformation = (_task ==
TaskKind
.Ranking) ?
252
var columnInformation = (_task ==
TaskKind
.Ranking) ?
API\MulticlassClassificationExperiment.cs (1)
136
TaskKind
.MulticlassClassification,
API\RankingExperiment.cs (1)
124
TaskKind
.Ranking,
API\RecommendationExperiment.cs (1)
63
TaskKind
.Recommendation,
API\RegressionExperiment.cs (1)
131
TaskKind
.Regression,
Experiment\Experiment.cs (2)
20
private readonly
TaskKind
_task;
41
TaskKind
task,
Experiment\RecipeInference.cs (1)
15
public static IEnumerable<SuggestedTrainer> AllowedTrainers(MLContext mlContext,
TaskKind
task,
PipelineSuggesters\PipelineSuggester.cs (2)
20
TaskKind
task,
32
TaskKind
task,
TrainerExtensions\TrainerExtensionCatalog.cs (7)
63
public static IEnumerable<ITrainerExtension> GetTrainers(
TaskKind
task,
67
if (task ==
TaskKind
.BinaryClassification)
71
else if (task ==
TaskKind
.MulticlassClassification &&
82
else if (task ==
TaskKind
.MulticlassClassification)
86
else if (task ==
TaskKind
.Regression)
90
else if (task ==
TaskKind
.Recommendation)
94
else if (task ==
TaskKind
.Ranking)
TransformInference\TransformInference.cs (15)
119
IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task);
124
public abstract IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task);
172
public override IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task)
174
if (task !=
TaskKind
.MulticlassClassification)
199
public override IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task)
201
if (task !=
TaskKind
.Recommendation)
222
public override IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task)
224
if (task !=
TaskKind
.Ranking)
244
public override IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task)
294
public override IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task)
322
public override IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task)
348
public override IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task)
376
public override IEnumerable<SuggestedTransform> Apply(IntermediateColumn[] columns,
TaskKind
task)
395
public static SuggestedTransform[] InferTransforms(MLContext context,
TaskKind
task, DatasetColumnInfo[] columns)
408
if (task !=
TaskKind
.Recommendation)
TransformInference\TransformInferenceApi.cs (2)
11
public static IEnumerable<SuggestedTransform> InferTransforms(MLContext context,
TaskKind
task, DatasetColumnInfo[] columns)
16
public static IEnumerable<SuggestedTransform> InferTransformsPostTrainer(MLContext context,
TaskKind
task, DatasetColumnInfo[] columns)
TransformInference\TransformPostTrainerInference.cs (3)
13
public static IEnumerable<SuggestedTransform> InferTransforms(MLContext context,
TaskKind
task, DatasetColumnInfo[] columns)
20
private static IEnumerable<SuggestedTransform> InferLabelTransforms(MLContext context,
TaskKind
task,
25
if (task !=
TaskKind
.MulticlassClassification)
Utils\UserInputValidationUtil.cs (10)
29
IDataView validationData,
TaskKind
task)
61
public static void ValidateSamplingKey(string samplingKeyColumnName, string groupIdColumnName,
TaskKind
task)
63
if (task ==
TaskKind
.Ranking && samplingKeyColumnName != null && samplingKeyColumnName != groupIdColumnName)
105
private static void ValidateColumnInformation(IDataView trainData, ColumnInformation columnInformation,
TaskKind
task)
307
private static IEnumerable<DataViewType> GetAllowedLabelTypes(
TaskKind
task)
311
case
TaskKind
.BinaryClassification:
315
case
TaskKind
.MulticlassClassification:
317
case
TaskKind
.Regression:
318
case
TaskKind
.Recommendation:
320
case
TaskKind
.Ranking:
Microsoft.ML.AutoML.Tests (86)
GetNextPipelineTests.cs (3)
32
TaskKind
.BinaryClassification, ((IChannelProvider)context).Start("AutoMLTest"));
57
var
task =
TaskKind
.BinaryClassification;
TrainerExtensionsTests.cs (2)
404
var trainers = RecipeInference.AllowedTrainers(new MLContext(1),
TaskKind
.BinaryClassification, new ColumnInformation(), null);
412
var trainers = RecipeInference.AllowedTrainers(new MLContext(1),
TaskKind
.BinaryClassification, new ColumnInformation(), allowList);
TransformInferenceTests.cs (5)
675
}, expectedJson, useRecommendationTask ?
TaskKind
.Recommendation :
TaskKind
.MulticlassClassification);
698
]",
TaskKind
.MulticlassClassification);
763
TaskKind
task =
TaskKind
.BinaryClassification)
TransformPostTrainerInferenceTests.cs (4)
24
TransformPostTrainerInferenceTestCore(
TaskKind
.MulticlassClassification,
47
TransformPostTrainerInferenceTestCore(
TaskKind
.BinaryClassification,
58
TransformPostTrainerInferenceTestCore(
TaskKind
.MulticlassClassification,
67
TaskKind
task,
UserInputValidationTests.cs (72)
29
var ex = Assert.Throws<ArgumentNullException>(() => UserInputValidationUtil.ValidateExperimentExecuteArgs(null, new ColumnInformation(), null,
TaskKind
.Regression));
37
new ColumnInformation() { LabelColumnName = null }, null,
TaskKind
.Regression));
45
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
58
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
76
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
89
var ex = Assert.Throws<ArgumentException>(() => UserInputValidationUtil.ValidateExperimentExecuteArgs(_data, columnInfo, null,
TaskKind
.Regression));
99
var ex = Assert.Throws<ArgumentException>(() => UserInputValidationUtil.ValidateExperimentExecuteArgs(_data, columnInfo, null,
TaskKind
.Regression));
116
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
139
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
162
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
165
new ColumnInformation() { LabelColumnName = "0" }, validData,
TaskKind
.Regression));
208
var ex = Assert.Throws<ArgumentException>(() => UserInputValidationUtil.ValidateExperimentExecuteArgs(dataView, new ColumnInformation(), null,
TaskKind
.Regression));
226
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
236
ValidateLabelTypeTestCore<float>(
TaskKind
.Regression, NumberDataViewType.Single, true);
237
ValidateLabelTypeTestCore<bool>(
TaskKind
.Regression, BooleanDataViewType.Instance, false);
238
ValidateLabelTypeTestCore<double>(
TaskKind
.Regression, NumberDataViewType.Double, false);
239
ValidateLabelTypeTestCore<string>(
TaskKind
.Regression, TextDataViewType.Instance, false);
245
ValidateLabelTypeTestCore<float>(
TaskKind
.Recommendation, NumberDataViewType.Single, true);
246
ValidateLabelTypeTestCore<bool>(
TaskKind
.Recommendation, BooleanDataViewType.Instance, false);
247
ValidateLabelTypeTestCore<double>(
TaskKind
.Recommendation, NumberDataViewType.Double, false);
248
ValidateLabelTypeTestCore<string>(
TaskKind
.Recommendation, TextDataViewType.Instance, false);
254
ValidateLabelTypeTestCore<float>(
TaskKind
.BinaryClassification, NumberDataViewType.Single, false);
255
ValidateLabelTypeTestCore<bool>(
TaskKind
.BinaryClassification, BooleanDataViewType.Instance, true);
261
ValidateLabelTypeTestCore<float>(
TaskKind
.MulticlassClassification, NumberDataViewType.Single, true);
262
ValidateLabelTypeTestCore<bool>(
TaskKind
.MulticlassClassification, BooleanDataViewType.Instance, true);
263
ValidateLabelTypeTestCore<double>(
TaskKind
.MulticlassClassification, NumberDataViewType.Double, true);
264
ValidateLabelTypeTestCore<string>(
TaskKind
.MulticlassClassification, TextDataViewType.Instance, true);
270
ValidateLabelTypeTestCore<float>(
TaskKind
.Ranking, NumberDataViewType.Single, true);
271
ValidateLabelTypeTestCore<bool>(
TaskKind
.Ranking, BooleanDataViewType.Instance, false);
272
ValidateLabelTypeTestCore<double>(
TaskKind
.Ranking, NumberDataViewType.Double, false);
273
ValidateLabelTypeTestCore<string>(
TaskKind
.Ranking, TextDataViewType.Instance, false);
286
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
303
null,
TaskKind
.Regression));
316
null,
TaskKind
.Regression));
336
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
368
UserInputValidationUtil.ValidateExperimentExecuteArgs(trainingData, new ColumnInformation(), validationData,
TaskKind
.BinaryClassification);
372
private static void ValidateLabelTypeTestCore<TLabelRawType>(
TaskKind
task, PrimitiveDataViewType labelType, bool labelTypeShouldBeValid)
399
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
414
foreach (
var
task in new[] {
TaskKind
.Recommendation,
TaskKind
.Regression,
TaskKind
.Ranking })
Microsoft.ML.CodeGenerator (1)
CodeGenerator\CSharp\CodeGeneratorSettings.cs (1)
67
internal
TaskKind
MlTask { get; set; }
Microsoft.ML.CodeGenerator.Tests (16)
ApprovalTests\ConsoleCodeGeneratorTests.cs (16)
56
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.MulticlassClassification));
71
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.BinaryClassification));
86
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.Regression));
101
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.Ranking));
116
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.BinaryClassification));
131
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.BinaryClassification));
146
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.BinaryClassification));
161
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.BinaryClassification));
176
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.BinaryClassification));
192
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.BinaryClassification));
207
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.BinaryClassification));
224
MlTask =
TaskKind
.MulticlassClassification,
260
MlTask =
TaskKind
.ObjectDetection,
300
MlTask =
TaskKind
.MulticlassClassification,
536
var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(
TaskKind
.Recommendation));
1130
private static CodeGeneratorSettings CreateCodeGeneratorSettingsFor(
TaskKind
task)