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