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.

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 @see …

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()