@Internal public class SGD extends Object implements Optimizer
See https://en.wikipedia.org/wiki/Stochastic_gradient_descent.
Constructor and Description |
---|
SGD(int maxIter,
double learningRate,
int globalBatchSize,
double tol,
double reg,
double elasticNet) |
Modifier and Type | Method and Description |
---|---|
org.apache.flink.streaming.api.datastream.DataStream<DenseVector> |
optimize(org.apache.flink.streaming.api.datastream.DataStream<DenseVector> initModelData,
org.apache.flink.streaming.api.datastream.DataStream<LabeledPointWithWeight> trainData,
LossFunc lossFunc)
Optimizes the given loss function using the initial model data and the bounded training data.
|
public SGD(int maxIter, double learningRate, int globalBatchSize, double tol, double reg, double elasticNet)
public org.apache.flink.streaming.api.datastream.DataStream<DenseVector> optimize(org.apache.flink.streaming.api.datastream.DataStream<DenseVector> initModelData, org.apache.flink.streaming.api.datastream.DataStream<LabeledPointWithWeight> trainData, LossFunc lossFunc)
Optimizer
Copyright © 2019–2023 The Apache Software Foundation. All rights reserved.