jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/optax
sgd
Function sgd
sgd
(
learningRate
:
number
|
(
(
count
:
number
)
=>
number
)
,
opts
?:
SgdOptions
,
)
:
GradientTransformation
Stochastic gradient descent.
Parameters
learningRate
:
number
|
(
(
count
:
number
)
=>
number
)
opts
:
SgdOptions
= {}
Returns
GradientTransformation
jax-js
Loading...
Stochastic gradient descent.