Package | Description |
---|---|
org.apache.flink.ml.common.lossfunc | |
org.apache.flink.ml.common.optimizer |
Modifier and Type | Method and Description |
---|---|
void |
HingeLoss.computeGradient(LabeledPointWithWeight dataPoint,
DenseVector coefficient,
DenseVector cumGradient) |
void |
LeastSquareLoss.computeGradient(LabeledPointWithWeight dataPoint,
DenseVector coefficient,
DenseVector cumGradient) |
void |
LossFunc.computeGradient(LabeledPointWithWeight dataPoint,
DenseVector coefficient,
DenseVector cumGradient)
Computes the gradient on the given data point and adds the computed gradient to cumGradient.
|
void |
BinaryLogisticLoss.computeGradient(LabeledPointWithWeight dataPoint,
DenseVector coefficient,
DenseVector cumGradient) |
double |
HingeLoss.computeLoss(LabeledPointWithWeight dataPoint,
DenseVector coefficient) |
double |
LeastSquareLoss.computeLoss(LabeledPointWithWeight dataPoint,
DenseVector coefficient) |
double |
LossFunc.computeLoss(LabeledPointWithWeight dataPoint,
DenseVector coefficient)
Computes the loss on the given data point.
|
double |
BinaryLogisticLoss.computeLoss(LabeledPointWithWeight dataPoint,
DenseVector coefficient) |
Modifier and Type | Method and Description |
---|---|
org.apache.flink.streaming.api.datastream.DataStream<DenseVector> |
Optimizer.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> |
SGD.optimize(org.apache.flink.streaming.api.datastream.DataStream<DenseVector> initModelData,
org.apache.flink.streaming.api.datastream.DataStream<LabeledPointWithWeight> trainData,
LossFunc lossFunc) |
Copyright © 2019–2023 The Apache Software Foundation. All rights reserved.