27 references to Regression
Microsoft.ML.AutoML (3)
API\RegressionExperiment.cs (1)
131TaskKind.Regression,
TrainerExtensions\TrainerExtensionCatalog.cs (1)
86else if (task == TaskKind.Regression)
Utils\UserInputValidationUtil.cs (1)
317case TaskKind.Regression:
Microsoft.ML.AutoML.Tests (23)
UserInputValidationTests.cs (23)
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); 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 }) 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.Tests (1)
ApprovalTests\ConsoleCodeGeneratorTests.cs (1)
86var consoleCodeGen = new CodeGenerator(pipeline, columnInference, CreateCodeGeneratorSettingsFor(TaskKind.Regression));