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)); }