-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Closed
Labels
P1Priority of the issue for triage purpose: Needs to be fixed soon.Priority of the issue for triage purpose: Needs to be fixed soon.bugSomething isn't workingSomething isn't workingclassificationBugs related classification tasksBugs related classification tasksonnxExporting ONNX models or loading ONNX modelsExporting ONNX models or loading ONNX models
Description
The following code fails to convert to ONNX model:
var mlContext = new MLContext();
var data = mlContext.Data.LoadFromEnumerable<Observation>(Data);
var chain = mlContext.Transforms.Concatenate("FeatureVector", "Price")
.Append(mlContext.BinaryClassification.Trainers.LogisticRegression(
"Label", "FeatureVector"));
var transformer = chain.Fit(data);
using (var stream = File.Create("foo.onnx"))
mlContext.Model.ConvertToOnnx(transformer, data, stream);
On the other hand, if you rename "FeatureVector" to "Features" it works.
This happens for various learners and both binary and multiclass. for both calibrated and non-calibrated. This does NOT happen on regression
Metadata
Metadata
Assignees
Labels
P1Priority of the issue for triage purpose: Needs to be fixed soon.Priority of the issue for triage purpose: Needs to be fixed soon.bugSomething isn't workingSomething isn't workingclassificationBugs related classification tasksBugs related classification tasksonnxExporting ONNX models or loading ONNX modelsExporting ONNX models or loading ONNX models