14 implementations of SaveAsOnnx
Microsoft.ML.Data (3)
Prediction\Calibrator.cs (3)
379
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumnName)
1294
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumn)
1792
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] scoreProbablityColumnNames, string featureColumnName)
Microsoft.ML.FastTree (3)
FastTree.cs (1)
3129
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumn)
FastTreeTweedie.cs (1)
535
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumn)
RandomForestRegression.cs (1)
218
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumn)
Microsoft.ML.KMeansClustering (1)
KMeansModelParameters.cs (1)
289
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumn)
Microsoft.ML.StandardTrainers (7)
Standard\LinearModelParameters.cs (2)
258
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputs, string featureColumn)
738
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputs, string featureColumn)
Standard\LogisticRegression\MulticlassLogisticRegression.cs (1)
1020
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputs, string featureColumn) => SaveAsOnnxCore(ctx, outputs, featureColumn);
Standard\MulticlassClassification\MulticlassNaiveBayesTrainer.cs (1)
422
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumn)
Standard\MulticlassClassification\OneVersusAllTrainer.cs (2)
498
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumn) => _impl.SaveAsOnnx(ctx, outputNames, featureColumn);
511
public abstract bool
SaveAsOnnx
(OnnxContext ctx, string[] outputNames, string featureColumn);
Standard\Simple\SimpleTrainers.cs (1)
403
bool ISingleCanSaveOnnx.
SaveAsOnnx
(OnnxContext ctx, string[] outputs, string labelColumn)
10 references to SaveAsOnnx
Microsoft.ML.Data (7)
Prediction\Calibrator.cs (3)
377
=> ((ISingleCanSaveOnnx)this).
SaveAsOnnx
(ctx, outputNames, featureColumnName);
387
if (!mapper.
SaveAsOnnx
(ctx, new[] { outputNames[1] }, featureColumnName))
391
if (!(calibrator?.CanSaveOnnx(ctx) == true && calibrator.
SaveAsOnnx
(ctx, new[] { outputNames[1], outputNames[2] }, featureColumnName)))
Prediction\CalibratorCatalog.cs (1)
276
onnx.
SaveAsOnnx
(ctx, new[] { scoreName, probabilityName }, ""); // No need for featureColumn
Scorers\SchemaBindablePredictorWrapper.cs (3)
352
return mapper.
SaveAsOnnx
(ctx, outputNames, ctx.GetVariableName(featName));
465
return mapper.
SaveAsOnnx
(ctx, outputNames, ctx.GetVariableName(labelColumnName));
472
return mapper.
SaveAsOnnx
(ctx, outputNames, ctx.GetVariableName(featName));
Microsoft.ML.StandardTrainers (3)
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.
Standard\MulticlassClassification\OneVersusAllTrainer.cs (1)
554
pred.
SaveAsOnnx
(ctx, predictorOutputNames, featureColumn);