6 references to TryGetCategoricalFeatureIndices
Microsoft.ML.Data (3)
Transforms\ColumnConcatenatingTransformer.cs (2)
495if (AnnotationUtils.TryGetCategoricalFeatureIndices(inputSchema, srcCol, out int[] typeCat)) 604if (AnnotationUtils.TryGetCategoricalFeatureIndices(_inputSchema, SrcIndices[i], out int[] values))
Transforms\SlotsDroppingTransformer.cs (1)
854if (AnnotationUtils.TryGetCategoricalFeatureIndices(InputSchema, _cols[iinfo], out _categoricalRanges[iinfo]))
Microsoft.ML.FastTree (2)
FastTree.cs (1)
190AnnotationUtils.TryGetCategoricalFeatureIndices(trainData.Schema.Schema, trainData.Schema.Feature.Value.Index, out CategoricalFeatures);
GamModelParameters.cs (1)
688AnnotationUtils.TryGetCategoricalFeatureIndices(_data.Schema.Schema, featureCol.Index, out _catsMap);
Microsoft.ML.LightGbm (1)
LightGbmTrainerBase.cs (1)
646AnnotationUtils.TryGetCategoricalFeatureIndices(trainData.Schema.Schema, featureCol.Index, out categoricalFeatures);