65 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 (12)
GetNextPipelineTests.cs (1)
57var task = TaskKind.BinaryClassification;
UserInputValidationTests.cs (11)
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 }) 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 }) 226foreach (var task in new[] { TaskKind.Recommendation, TaskKind.Regression, TaskKind.Ranking }) 286foreach (var task in new[] { TaskKind.Recommendation, TaskKind.Regression, TaskKind.Ranking }) 336foreach (var task in new[] { TaskKind.Recommendation, TaskKind.Regression, TaskKind.Ranking }) 399foreach (var task in new[] { TaskKind.Recommendation, TaskKind.Regression, TaskKind.Ranking }) 414foreach (var task in new[] { TaskKind.Recommendation, TaskKind.Regression, TaskKind.Ranking })