@Internal
public interface Optimizer
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.
|
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)
initModelData
- The initial model data.trainData
- The training data.lossFunc
- The loss function to optimize.Copyright © 2019–2023 The Apache Software Foundation. All rights reserved.