Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions src/Microsoft.ML.AutoML/CodeGen/search-space-schema.json
Original file line number Diff line number Diff line change
Expand Up @@ -217,7 +217,8 @@
"TrainingAnswerColumnName",
"AnswerIndexStartColumnName",
"predictedAnswerColumnName",
"TopKAnswers"
"TopKAnswers",
"TargetType"
]
},
"option_type": {
Expand All @@ -235,7 +236,8 @@
"anchor",
"dnnModelFactory",
"bertArchitecture",
"imageClassificationArchType"
"imageClassificationArchType",
"DataKind"
]
}
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,10 @@
{
"name": "InputColumnNames",
"type": "strings"
},
{
"name": "TargetType",
"type": "DataKind"
}
]
}
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ internal partial class ConvertType
public override IEstimator<ITransformer> BuildFromOption(MLContext context, ConvertTypeOption param)
{
var inputOutputPairs = AutoMlUtils.CreateInputOutputColumnPairsFromStrings(param.InputColumnNames, param.OutputColumnNames);
return context.Transforms.Conversion.ConvertType(inputOutputPairs);
return context.Transforms.Conversion.ConvertType(inputOutputPairs, param.TargetType);
}
}
}