internal static BinaryClassificationResult AutoFit(this BinaryClassificationContext context,
IDataView trainData,
string label,
IDataView validationData = null,
InferredColumn[] inferredColumns = null,
AutoFitSettings settings = null,
CancellationToken cancellationToken = default,
IProgress <BinaryClassificationItertionResult> 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.BinaryClassification, OptimizingMetric.Accuracy,
debugLogger);
var results = new BinaryClassificationItertionResult[allPipelines.Length];
for (var i = 0; i < results.Length; i++)
{
var iterationResult = allPipelines[i];
var result = new BinaryClassificationItertionResult(iterationResult.Model, (BinaryClassificationMetrics)iterationResult.EvaluatedMetrics, iterationResult.ScoredValidationData);
results[i] = result;
}
var bestResult = new BinaryClassificationItertionResult(bestPipeline.Model, (BinaryClassificationMetrics)bestPipeline.EvaluatedMetrics, bestPipeline.ScoredValidationData);
return(new BinaryClassificationResult(bestResult, results));
}
public BinaryClassificationResult(BinaryClassificationItertionResult bestPipeline,
BinaryClassificationItertionResult[] iterationResults)
{
BestPipeline = bestPipeline;
IterationResults = iterationResults;
}
