public PipelineNode CreatePipelineNode(IEnumerable <SweepableParam> sweepParams, ColumnInformation columnInfo)
{
var property = new Dictionary <string, object>();
property.Add(nameof(FastTreeRankingTrainer.Options.RowGroupColumnName), columnInfo.GroupIdColumnName);
return(TrainerExtensionUtil.BuildPipelineNode(TrainerExtensionCatalog.GetTrainerName(this), sweepParams,
columnInfo.LabelColumnName, additionalProperties: property));
}
public PipelineNode CreatePipelineNode(IEnumerable <SweepableParam> sweepParams, ColumnInformation columnInfo)
{
var property = new Dictionary <string, object>();
property.Add(nameof(MatrixFactorizationTrainer.Options.MatrixColumnIndexColumnName), columnInfo.UserIdColumnName);
property.Add(nameof(MatrixFactorizationTrainer.Options.MatrixRowIndexColumnName), columnInfo.ItemIdColumnName);
return(TrainerExtensionUtil.BuildPipelineNode(TrainerExtensionCatalog.GetTrainerName(this), sweepParams, columnInfo.LabelColumnName, additionalProperties: property));
}
public PipelineNode CreatePipelineNode(IEnumerable <SweepableParam> sweepParams, ColumnInformation columnInfo)
{
Dictionary <string, object> additionalProperties = null;
if (sweepParams == null || !sweepParams.Any(p => p.Name != "NumberOfIterations"))
{
additionalProperties = new Dictionary <string, object>()
{
{ "NumberOfIterations", DefaultNumIterations }
};
}
return(TrainerExtensionUtil.BuildPipelineNode(TrainerExtensionCatalog.GetTrainerName(this), sweepParams,
columnInfo.LabelColumnName, additionalProperties: additionalProperties));
}
public PipelineNode CreatePipelineNode(IEnumerable <SweepableParam> sweepParams, ColumnInformation columnInfo)
{
return(TrainerExtensionUtil.BuildPipelineNode(TrainerExtensionCatalog.GetTrainerName(this), sweepParams,
columnInfo.LabelColumnName, null));
}