2 writes to NodeType
Microsoft.ML.AutoML (2)
API\Pipeline.cs (1)
45NodeType = nodeType;
TrainerExtensions\TrainerExtensionUtil.cs (1)
106NodeType = PipelineNodeType.Trainer,
13 references to NodeType
Microsoft.ML.AutoML (2)
Experiment\SuggestedPipeline.cs (2)
84if (pipelineNode.NodeType == PipelineNodeType.Trainer) 93else if (pipelineNode.NodeType == PipelineNodeType.Transform)
Microsoft.ML.CodeGenerator (11)
CodeGenerator\CSharp\CodeGenerator.cs (6)
46var trainerNodes = _pipeline.Nodes.Where(t => t.NodeType == PipelineNodeType.Trainer); 204var nodes = _pipeline.Nodes.TakeWhile(t => t.NodeType == PipelineNodeType.Transform); 208nodes = _pipeline.Nodes.SkipWhile(t => t.NodeType == PipelineNodeType.Transform) 209.SkipWhile(t => t.NodeType == PipelineNodeType.Trainer) //skip the trainer 210.TakeWhile(t => t.NodeType == PipelineNodeType.Transform); //post trainer transforms 256var node = _pipeline.Nodes.Where(t => t.NodeType == PipelineNodeType.Trainer).First();
CodeGenerator\CSharp\PipelineExtension.cs (5)
22var nodes = pipeline.Nodes.TakeWhile(t => t.NodeType == PipelineNodeType.Transform); 26nodes = pipeline.Nodes.SkipWhile(t => t.NodeType == PipelineNodeType.Transform) 27.SkipWhile(t => t.NodeType == PipelineNodeType.Trainer) //skip the trainer 28.TakeWhile(t => t.NodeType == PipelineNodeType.Transform); //post trainer transforms 76var node = pipeline.Nodes.Where(t => t.NodeType == PipelineNodeType.Trainer).First();