Optimizer

public protocol Optimizer

A numerical optimizer.

Optimizers apply an optimization algorithm to update the differentiable models.

Model
The type of the model whose parameters are optimized.
Declaration
associatedtype Model : Differentiable
Scalar
The scalar parameter type.
Declaration
associatedtype Scalar : FloatingPoint
learningRate
The learning rate.
Declaration
var learningRate: Scalar { get set }
update(_:along:)
Updates the specified differentiable variables along the specified direction.
Declaration
mutating func update(_ variables: inout Model, along direction: Model.TangentVector)

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.