// todo: instead of internal methods, use static debug class w/ singleton logger? internal static RegressionResult AutoFit(this RegressionContext context, IDataView trainData, string label, IDataView validationData = null, InferredColumn[] inferredColumns = null, AutoFitSettings settings = null, CancellationToken cancellationToken = default, IProgress <RegressionIterationResult> iterationCallback = null, IDebugLogger debugLogger = null) { // run autofit & get all pipelines run in that process var(allPipelines, bestPipeline) = AutoFitApi.Fit(trainData, validationData, label, inferredColumns, settings, TaskKind.Regression, OptimizingMetric.RSquared, debugLogger); var results = new RegressionIterationResult[allPipelines.Length]; for (var i = 0; i < results.Length; i++) { var iterationResult = allPipelines[i]; var result = new RegressionIterationResult(iterationResult.Model, (RegressionMetrics)iterationResult.EvaluatedMetrics, iterationResult.ScoredValidationData); results[i] = result; } var bestResult = new RegressionIterationResult(bestPipeline.Model, (RegressionMetrics)bestPipeline.EvaluatedMetrics, bestPipeline.ScoredValidationData); return(new RegressionResult(bestResult, results)); }
public RegressionResult(RegressionIterationResult bestPipeline, RegressionIterationResult[] iterationResults) { BestPipeline = bestPipeline; IterationResults = iterationResults; }