3 implementations of SaveAsOnnx
Microsoft.ML.Data (3)
Prediction\Calibrator.cs (3)
379bool ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, string[] outputNames, string featureColumnName) 1294bool ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, string[] outputNames, string featureColumn) 1792bool ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, string[] scoreProbablityColumnNames, string featureColumnName)
9 references to SaveAsOnnx
Microsoft.ML.Data (7)
Prediction\Calibrator.cs (3)
377=> ((ISingleCanSaveOnnx)this).SaveAsOnnx(ctx, outputNames, featureColumnName); 387if (!mapper.SaveAsOnnx(ctx, new[] { outputNames[1] }, featureColumnName)) 391if (!(calibrator?.CanSaveOnnx(ctx) == true && calibrator.SaveAsOnnx(ctx, new[] { outputNames[1], outputNames[2] }, featureColumnName)))
Prediction\CalibratorCatalog.cs (1)
276onnx.SaveAsOnnx(ctx, new[] { scoreName, probabilityName }, ""); // No need for featureColumn
Scorers\SchemaBindablePredictorWrapper.cs (3)
352return mapper.SaveAsOnnx(ctx, outputNames, ctx.GetVariableName(featName)); 465return mapper.SaveAsOnnx(ctx, outputNames, ctx.GetVariableName(labelColumnName)); 472return mapper.SaveAsOnnx(ctx, outputNames, ctx.GetVariableName(featName));
Microsoft.ML.StandardTrainers (2)
Standard\LogisticRegression\MulticlassLogisticRegression.cs (2)
974/// Actual implementation of <see cref="ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext, string[], string)"/> should happen in derived classes. 1018/// Redirect <see cref="ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext, string[], string)"/> call to the right function.