public ITrainingService CreateTrainingService(SingleEstimatorSweepablePipeline pipeline)
{
if (!this.option.IsAzureAttach && !this.option.IsNNITraining)
{
var option = new LocalTrainingService.Option()
{
ParameterSweeper = this.option.ParameterSweeper,
EvaluationMetric = this.option.EvaluationMetric,
IsMaximizng = this.option.IsMaximizng,
Metrics = this.option.Metrics,
ParameterSweepingIteration = this.option.ParameterSweepingIteration,
Label = this.option.Label,
};
return(new LocalTrainingService(this.context, pipeline, option));
}
throw new Exception("Not implemented");
}
public static SingleEstimatorSweepablePipelineDataContract ToDataContract(this SingleEstimatorSweepablePipeline pipeline)
{
var estimatorContracts = new List <SweepableEstimatorDataContract>();
var nodes = pipeline.Estimators;
foreach (var node in nodes)
{
var estimatorContract = new SweepableEstimatorDataContract()
{
EstimatorName = node.EstimatorName,
InputColumns = node.InputColumns,
OutputColumns = node.OutputColumns,
Scope = node.Scope,
};
estimatorContracts.Add(estimatorContract);
}
return(new SingleEstimatorSweepablePipelineDataContract()
{
Estimators = estimatorContracts,
});
}
public LocalTrainingService(MLContext context, SingleEstimatorSweepablePipeline pipeline, Option option)
{
this.context = context;
this.option = option;
this.Pipeline = pipeline;
}
