algorithm property
The algorithm used to train the MLModel
. The following
algorithm is supported:
-
SGD
-- Stochastic gradient descent. The goal ofSGD
is to minimize the gradient of the loss function.
Implementation
final Algorithm? algorithm;
The algorithm used to train the MLModel
. The following
algorithm is supported:
SGD
-- Stochastic gradient descent. The goal of
SGD
is to minimize the gradient of the loss function. final Algorithm? algorithm;