接口 Optimizer
-
- 所有已知实现类:
SGD
@Internal public interface OptimizerAn optimizer is a function to modify the weight of a machine learning model, which aims to find the optimal parameter configuration for a machine learning model. Examples of optimizers could be stochastic gradient descent (SGD), L-BFGS, etc.
-
-
方法概要
所有方法 实例方法 抽象方法 修饰符和类型 方法 说明 org.apache.flink.streaming.api.datastream.DataStream<org.apache.flink.ml.linalg.DenseVector>optimize(org.apache.flink.streaming.api.datastream.DataStream<org.apache.flink.ml.linalg.DenseVector> initModelData, org.apache.flink.streaming.api.datastream.DataStream<org.apache.flink.ml.common.feature.LabeledPointWithWeight> trainData, LossFunc lossFunc)Optimizes the given loss function using the initial model data and the bounded training data.
-
-
-
方法详细资料
-
optimize
org.apache.flink.streaming.api.datastream.DataStream<org.apache.flink.ml.linalg.DenseVector> optimize(org.apache.flink.streaming.api.datastream.DataStream<org.apache.flink.ml.linalg.DenseVector> initModelData, org.apache.flink.streaming.api.datastream.DataStream<org.apache.flink.ml.common.feature.LabeledPointWithWeight> trainData, LossFunc lossFunc)Optimizes the given loss function using the initial model data and the bounded training data.- 参数:
initModelData- The initial model data.trainData- The training data.lossFunc- The loss function to optimize.- 返回:
- The fitted model data.
-
-