59 references to EstimatorName
Microsoft.ML.AutoML (34)
EstimatorExtensions\EstimatorExtensionCatalog.cs (17)
30private static readonly IDictionary<EstimatorName, Type> _namesToExtensionTypes = new 31Dictionary<EstimatorName, Type>() 33{ EstimatorName.ColumnConcatenating, typeof(ColumnConcatenatingExtension) }, 34{ EstimatorName.ColumnCopying, typeof(ColumnCopyingExtension) }, 35{ EstimatorName.KeyToValueMapping, typeof(KeyToValueMappingExtension) }, 36{ EstimatorName.Hashing, typeof(HashingExtension) }, 37{ EstimatorName.MissingValueIndicating, typeof(MissingValueIndicatingExtension) }, 38{ EstimatorName.MissingValueReplacing, typeof(MissingValueReplacingExtension) }, 39{ EstimatorName.Normalizing, typeof(NormalizingExtension) }, 40{ EstimatorName.OneHotEncoding, typeof(OneHotEncodingExtension) }, 41{ EstimatorName.OneHotHashEncoding, typeof(OneHotHashEncodingExtension) }, 42{ EstimatorName.TextFeaturizing, typeof(TextFeaturizingExtension) }, 43{ EstimatorName.TypeConverting, typeof(TypeConvertingExtension) }, 44{ EstimatorName.ValueToKeyMapping, typeof(ValueToKeyMappingExtension) }, 45{ EstimatorName.RawByteImageLoading, typeof(RawByteImageLoading) }, 46{ EstimatorName.ImageLoading, typeof(ImageLoading) } 49public static IEstimatorExtension GetExtension(EstimatorName estimatorName)
EstimatorExtensions\EstimatorExtensions.cs (14)
18var pipelineNode = new PipelineNode(EstimatorName.ColumnConcatenating.ToString(), 39var pipelineNode = new PipelineNode(EstimatorName.ColumnCopying.ToString(), 60var pipelineNode = new PipelineNode(EstimatorName.KeyToValueMapping.ToString(), 81var pipelineNode = new PipelineNode(EstimatorName.Hashing.ToString(), 102var pipelineNode = new PipelineNode(EstimatorName.MissingValueIndicating.ToString(), 129var pipelineNode = new PipelineNode(EstimatorName.MissingValueReplacing.ToString(), 156var pipelineNode = new PipelineNode(EstimatorName.Normalizing.ToString(), 177var pipelineNode = new PipelineNode(EstimatorName.OneHotEncoding.ToString(), 208var pipelineNode = new PipelineNode(EstimatorName.OneHotHashEncoding.ToString(), 234var pipelineNode = new PipelineNode(EstimatorName.TextFeaturizing.ToString(), 255var pipelineNode = new PipelineNode(EstimatorName.TypeConverting.ToString(), 281var pipelineNode = new PipelineNode(EstimatorName.ValueToKeyMapping.ToString(), 308var pipelineNode = new PipelineNode(EstimatorName.RawByteImageLoading.ToString(), PipelineNodeType.Transform, inColumn, outColumn, pipelineNodeProperty); 333var pipelineNode = new PipelineNode(EstimatorName.ImageLoading.ToString(), PipelineNodeType.Transform, inColumn, outColumn, pipelineNodeProperty);
Experiment\SuggestedPipeline.cs (3)
95var estimatorName = (EstimatorName)Enum.Parse(typeof(EstimatorName), pipelineNode.Name);
Microsoft.ML.AutoML.Tests (4)
EstimatorExtensionTests.cs (3)
32var estimatorNames = Enum.GetValues(typeof(EstimatorName)).Cast<EstimatorName>(); 33foreach (var estimatorName in estimatorNames)
SuggestedPipelineBuilderTests.cs (1)
41Assert.Equal(EstimatorName.Normalizing.ToString(),
Microsoft.ML.CodeGenerator (16)
CodeGenerator\CSharp\CodeGenerator.cs (1)
33private static readonly HashSet<string> _imageTransformers = new HashSet<string>() { EstimatorName.RawByteImageLoading.ToString() };
CodeGenerator\CSharp\TransformGeneratorFactory.cs (15)
31if (Enum.TryParse(node.Name, out EstimatorName trainer)) 35case EstimatorName.Normalizing: 38case EstimatorName.OneHotEncoding: 41case EstimatorName.ColumnConcatenating: 44case EstimatorName.ColumnCopying: 47case EstimatorName.KeyToValueMapping: 50case EstimatorName.Hashing: 53case EstimatorName.MissingValueIndicating: 56case EstimatorName.MissingValueReplacing: 59case EstimatorName.OneHotHashEncoding: 62case EstimatorName.TextFeaturizing: 65case EstimatorName.TypeConverting: 68case EstimatorName.ValueToKeyMapping: 71case EstimatorName.RawByteImageLoading: 74case EstimatorName.ImageLoading:
Microsoft.ML.CodeGenerator.Tests (5)
ApprovalTests\ConsoleCodeGeneratorTests.cs (5)
558var valueToKeyPipelineNode1 = new PipelineNode(nameof(EstimatorName.ValueToKeyMapping), PipelineNodeType.Transform, "userId", "userId"); 559var valueToKeyPipelineNode2 = new PipelineNode(nameof(EstimatorName.ValueToKeyMapping), PipelineNodeType.Transform, "movieId", "movieId"); 678var loadImageNode = new PipelineNode(EstimatorName.ImageLoading.ToString(), PipelineNodeType.Transform, "ImagePath", "input"); 728var hashPipelineNode = new PipelineNode(nameof(EstimatorName.Hashing), PipelineNodeType.Transform, "GroupId", "GroupId"); 766var loadImageNode = new PipelineNode(EstimatorName.ImageLoading.ToString(), PipelineNodeType.Transform, "ImageSource", "ImageSource_featurized");