Package | Description |
---|---|
org.apache.flink.ml.optimization |
Modifier and Type | Class and Description |
---|---|
class |
GenericLossFunction
Generic loss function which lets you build a loss function out of the
PartialLossFunction
and the PredictionFunction . |
Modifier and Type | Method and Description |
---|---|
DataSet<WeightVector> |
GradientDescent.optimizeWithConvergenceCriterion(DataSet<LabeledVector> dataPoints,
DataSet<WeightVector> initialWeightsDS,
int numberOfIterations,
double regularizationConstant,
double learningRate,
double convergenceThreshold,
LossFunction lossFunction,
LearningRateMethod.LearningRateMethodTrait learningRateMethod) |
DataSet<WeightVector> |
GradientDescent.optimizeWithoutConvergenceCriterion(DataSet<LabeledVector> data,
DataSet<WeightVector> initialWeightsDS,
int numberOfIterations,
double regularizationConstant,
double learningRate,
LossFunction lossFunction,
LearningRateMethod.LearningRateMethodTrait optimizationMethod) |
Solver |
Solver.setLossFunction(LossFunction lossFunction) |
Copyright © 2014–2017 The Apache Software Foundation. All rights reserved.