Classes#

Summary#

class

class parent

truncated documentation

AbsoluteLearningLoss

Implements a square loss |Y - Z| where Y is the output and Z the expected output. See _onnx_grad_loss_absolute_error()

BaseEstimator

Base class for optimizers. Implements common methods such __repr__.

BaseLearningLoss

Class handling the loss for class OrtGradientForwardBackwardOptimizer. All classes inheriting from this …

BaseLearningOnnx

Class handling ONNX function to manipulate OrtValue. Base class for BaseLearningRate and BaseLearningLoss. …

BaseLearningPenalty

Class handling the penalty on the coefficients for class OrtGradientForwardBackwardOptimizer.

BaseLearningRate

Class handling the learning rate update after every iteration of a gradient. Two methods need to be overwritten …

BaseOnnxClass

Bases class with common functions to handle attributes in classes owning ONNX graphs.

Cbids

List of events.

ConvergenceError

Raised when a learning algorithm failed to converge.

ConvergenceWarning

Raised when a learning algorithm failed to converge.

DerivativeOptions

Options defining how to build the onnx graph of the gradients.

ElasticLearningLoss

Implements a square loss (Y - Z)^2 \alpha + |Y - Z| * \beta where Y is the output and Z the expected …

ElasticLearningPenalty

Implements a L1 or L2 regularization on weights.

EvaluationError

Raised when an evaluation failed.

LearningRateSGD

Implements the learning the same way as sklearn.linear_model.SGDRegressor.

LearningRateSGDNesterov

Implements the learning the same way as sklearn.linear_model.SGDRegressor.

NegLogLearningLoss

Implements a negative log loss ‘log(yt, yp) = -(1-yt)log(1-yp) - ytlog(yp), this only works for a binary classification …

NoLearningPenalty

No regularization.

OnnxSegment

A segments of an onnx graph assuming it is the concatenation of all segments.

OnnxSplitting

The final goal is to split an onnx model into equivalent pieces.

OrtDataLoader

Draws consecutive random observations from a dataset by batch. It iterates over the datasets by drawing batch_size

OrtGradientForwardBackward

Implements forward backward mechanism assuming the function to train is defined by an ONNX graph.

OrtGradientForwardBackwardFunction

Ancestor for a class implementing forward and backward and dynamically created by OrtGradientForwardBackward. …

OrtGradientForwardBackwardOptimizer

Implements a simple Stochastic Gradient Descent with onnxruntime-training. It leverages class OrtGradientForwardBackward. …

OrtGradientOptimizer

Implements a simple Stochastic Gradient Descent with onnxruntime-training.

ProviderError

Raised when an input is not on the expected device (CPU, GPU).

SquareLearningLoss

Implements a square loss (Y - Z)^2 where Y is the output and Z the expected output. See _onnx_grad_loss_square_error()