internal (string, string[]) GenerateTrainerAndUsings()
{
if (_pipeline == null)
{
throw new ArgumentNullException(nameof(_pipeline));
}
var node = _pipeline.Nodes.Where(t => t.NodeType == PipelineNodeType.Trainer).First();
ITrainerGenerator generator = TrainerGeneratorFactory.GetInstance(node);
var trainerString = generator.GenerateTrainer();
var trainerUsings = generator.GenerateUsings();
return(trainerString, trainerUsings);
}
public override string GenerateTrainer()
{
StringBuilder sb = new StringBuilder();
sb.Append(MethodName);
sb.Append("(");
sb.Append("mlContext.BinaryClassification.Trainers."); // This is dependent on the name of the MLContext object in template.
var trainerGenerator = TrainerGeneratorFactory.GetInstance((PipelineNode)_node.Properties["BinaryTrainer"]);
_binaryTrainerUsings = trainerGenerator.GenerateUsings();
sb.Append(trainerGenerator.GenerateTrainer());
sb.Append(",");
sb.Append("labelColumnName:");
sb.Append("\"");
sb.Append(_node.Properties["LabelColumnName"]);
sb.Append("\"");
sb.Append(")");
return(sb.ToString());
}
