Skip to content

Commit 9915e47

Browse files
add addKeyValueAnnotationsAsText flag
1 parent bc250df commit 9915e47

File tree

3 files changed

+8
-2
lines changed

3 files changed

+8
-2
lines changed

src/Microsoft.ML.AutoML/CodeGen/map_value_to_key_search_space.json

+5
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,11 @@
99
{
1010
"name": "InputColumnName",
1111
"type": "string"
12+
},
13+
{
14+
"name": "AddKeyValueAnnotationsAsText",
15+
"type": "boolean",
16+
"default": false
1217
}
1318
]
1419
}

src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json

+2-1
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,8 @@
177177
"Sentence2ColumnName",
178178
"BatchSize",
179179
"MaxEpochs",
180-
"Architecture"
180+
"Architecture",
181+
"AddKeyValueAnnotationsAsText"
181182
]
182183
},
183184
"option_type": {

src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/MapValueToKey.cs

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ internal partial class MapValueToKey
88
{
99
public override IEstimator<ITransformer> BuildFromOption(MLContext context, MapValueToKeyOption param)
1010
{
11-
return context.Transforms.Conversion.MapValueToKey(param.OutputColumnName, param.InputColumnName);
11+
return context.Transforms.Conversion.MapValueToKey(param.OutputColumnName, param.InputColumnName, addKeyValueAnnotationsAsText: param.AddKeyValueAnnotationsAsText);
1212
}
1313
}
1414

0 commit comments

Comments
 (0)