26 references to CategoricalSlotRanges
Microsoft.ML.Core (2)
Data\AnnotationUtils.cs (2)
388
var type = schema[colIndex].Annotations.Schema.GetColumnOrNull(Kinds.
CategoricalSlotRanges
)?.Type;
392
schema[colIndex].Annotations.GetValue(Kinds.
CategoricalSlotRanges
, ref catIndices);
Microsoft.ML.Data (6)
Transforms\ColumnConcatenatingEstimator.cs (2)
78
if (!col.Annotations.TryFindColumn(AnnotationUtils.Kinds.
CategoricalSlotRanges
, out var mcol))
134
meta.Add(new SchemaShape.Column(AnnotationUtils.Kinds.
CategoricalSlotRanges
, SchemaShape.Column.VectorKind.Vector, NumberDataViewType.Int32, false));
Transforms\ColumnConcatenatingTransformer.cs (1)
585
metadata.Add(AnnotationUtils.Kinds.
CategoricalSlotRanges
, _categoricalRangeType, (ValueGetter<VBuffer<int>>)GetCategoricalSlotRanges);
Transforms\KeyToVector.cs (2)
350
builder.Add(AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.GetCategoricalType(srcValueCount), getter);
834
metadata.Add(new SchemaShape.Column(AnnotationUtils.Kinds.
CategoricalSlotRanges
, SchemaShape.Column.VectorKind.Vector, NumberDataViewType.Int32, false));
Transforms\SlotsDroppingTransformer.cs (1)
864
builder.Add(AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.GetCategoricalType(dst.Length / 2), categoricalSlotRangesGetter);
Microsoft.ML.Tests (16)
Transformers\CategoricalHashTests.cs (4)
169
Assert.Equal(column.Annotations.Schema.Select(x => x.Name), new string[3] { AnnotationUtils.Kinds.SlotNames, AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.Kinds.IsNormalized });
175
column.Annotations.GetValue(AnnotationUtils.Kinds.
CategoricalSlotRanges
, ref slotRanges);
181
Assert.Equal(column.Annotations.Schema.Select(x => x.Name), new string[3] { AnnotationUtils.Kinds.SlotNames, AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.Kinds.IsNormalized });
185
column.Annotations.GetValue(AnnotationUtils.Kinds.
CategoricalSlotRanges
, ref slotRanges);
Transformers\CategoricalTests.cs (4)
247
Assert.Equal(column.Annotations.Schema.Select(x => x.Name), new string[3] { AnnotationUtils.Kinds.SlotNames, AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.Kinds.IsNormalized });
251
column.Annotations.GetValue(AnnotationUtils.Kinds.
CategoricalSlotRanges
, ref slotRanges);
257
Assert.Equal(column.Annotations.Schema.Select(x => x.Name), new string[3] { AnnotationUtils.Kinds.SlotNames, AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.Kinds.IsNormalized });
261
column.Annotations.GetValue(AnnotationUtils.Kinds.
CategoricalSlotRanges
, ref slotRanges);
Transformers\KeyToVectorEstimatorTests.cs (8)
142
Assert.Equal(column.Annotations.Schema.Select(x => x.Name), new string[3] { AnnotationUtils.Kinds.SlotNames, AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.Kinds.IsNormalized });
146
column.Annotations.GetValue(AnnotationUtils.Kinds.
CategoricalSlotRanges
, ref slotRanges);
152
Assert.Equal(column.Annotations.Schema.Select(x => x.Name), new string[3] { AnnotationUtils.Kinds.SlotNames, AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.Kinds.IsNormalized });
156
column.Annotations.GetValue(AnnotationUtils.Kinds.
CategoricalSlotRanges
, ref slotRanges);
169
Assert.Equal(column.Annotations.Schema.Select(x => x.Name), new string[2] { AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.Kinds.IsNormalized });
170
column.Annotations.GetValue(AnnotationUtils.Kinds.
CategoricalSlotRanges
, ref slotRanges);
186
Assert.Equal(column.Annotations.Schema.Select(x => x.Name), new string[3] { AnnotationUtils.Kinds.SlotNames, AnnotationUtils.Kinds.
CategoricalSlotRanges
, AnnotationUtils.Kinds.IsNormalized });
190
column.Annotations.GetValue(AnnotationUtils.Kinds.
CategoricalSlotRanges
, ref slotRanges);
Microsoft.ML.Transforms (2)
CountFeatureSelection.cs (1)
179
if (col.Annotations.TryFindColumn(AnnotationUtils.Kinds.
CategoricalSlotRanges
, out var categoricalSlotMeta))
MutualInformationFeatureSelection.cs (1)
246
if (col.Annotations.TryFindColumn(AnnotationUtils.Kinds.
CategoricalSlotRanges
, out var categoricalSlotMeta))