public PipelineNode ToPipelineNode()
{
var hyperParams = SweepParams.Where(p => p != null && p.RawValue != null);
var elementProperties = new Dictionary <string, object>();
foreach (var hyperParam in hyperParams)
{
elementProperties[hyperParam.Name] = hyperParam.ProcessedValue();
}
return(new PipelineNode(TrainerName.ToString(), PipelineNodeType.Trainer,
new[] { "Features" }, new[] { "Score" }, elementProperties));
}
public static ParameterSet BuildParameterSet(TrainerName trainerName, IDictionary <string, object> props)
{
props = props.Where(p => p.Key != LabelColumn && p.Key != WeightColumn)
.ToDictionary(kvp => kvp.Key, kvp => kvp.Value);
if (trainerName == TrainerName.LightGbmBinary || trainerName == TrainerName.LightGbmMulti ||
trainerName == TrainerName.LightGbmRegression || trainerName == TrainerName.LightGbmRanking)
{
return(BuildLightGbmParameterSet(props));
}
var paramVals = props.Select(p => new StringParameterValue(p.Key, p.Value.ToString()));
return(new ParameterSet(paramVals));
}
public static PipelineNode BuildPipelineNode(TrainerName trainerName, IEnumerable <SweepableParam> sweepParams,
string labelColumn, string weightColumn = null, IDictionary <string, object> additionalProperties = null)
{
var properties = BuildBasePipelineNodeProps(sweepParams, labelColumn, weightColumn);
if (additionalProperties != null)
{
foreach (var property in additionalProperties)
{
properties[property.Key] = property.Value;
}
}
return(new PipelineNode(trainerName.ToString(), PipelineNodeType.Trainer, DefaultColumnNames.Features,
DefaultColumnNames.Score, properties));
}
public static PipelineNode BuildLightGbmPipelineNode(TrainerName trainerName, IEnumerable <SweepableParam> sweepParams,
string labelColumn, string weightColumn, string groupColumn)
{
return(new PipelineNode(trainerName.ToString(), PipelineNodeType.Trainer, DefaultColumnNames.Features,
DefaultColumnNames.Score, BuildLightGbmPipelineNodeProps(sweepParams, labelColumn, weightColumn, groupColumn)));
}
public static ITrainerExtension GetTrainerExtension(TrainerName trainerName)
{
var trainerExtensionType = _trainerNamesToExtensionTypes[trainerName];
return((ITrainerExtension)Activator.CreateInstance(trainerExtensionType));
}
