internal SuggestedTrainer(MLContext mlContext, ITrainerExtension trainerExtension,
ParameterSet hyperParamSet = null)
{
_mlContext = mlContext;
_trainerExtension = trainerExtension;
SweepParams = _trainerExtension.GetHyperparamSweepRanges();
TrainerName = _trainerExtension.GetTrainerName();
SetHyperparamValues(hyperParamSet);
}
internal SuggestedTrainer(MLContext mlContext, ITrainerExtension trainerExtension,
ColumnInformation columnInfo,
ParameterSet hyperParamSet = null)
{
_mlContext = mlContext;
_trainerExtension = trainerExtension;
_columnInfo = columnInfo;
SweepParams = _trainerExtension.GetHyperparamSweepRanges();
TrainerName = TrainerExtensionCatalog.GetTrainerName(_trainerExtension);
SetHyperparamValues(hyperParamSet);
}
public static PipelineNode BuildOvaPipelineNode(ITrainerExtension multiExtension, ITrainerExtension binaryExtension,
IEnumerable <SweepableParam> sweepParams, ColumnInformation columnInfo)
{
var ovaNode = new PipelineNode()
{
Name = TrainerName.Ova.ToString(),
NodeType = PipelineNodeType.Trainer,
Properties = new Dictionary <string, object>()
{
{ LabelColumn, columnInfo.LabelColumnName }
}
};
var binaryNode = binaryExtension.CreatePipelineNode(sweepParams, columnInfo);
ovaNode.Properties["BinaryTrainer"] = binaryNode;
return(ovaNode);
}
public static TrainerName GetTrainerName(ITrainerExtension trainerExtension)
{
return(_extensionTypesToTrainerNames[trainerExtension.GetType()]);
}
