public class StochasticGradientDescent extends Object implements StochasticMinimizer
| Modifier and Type | Class and Description |
|---|---|
static class |
StochasticGradientDescent.StochasticGradientDescentBuilder |
| Modifier and Type | Method and Description |
|---|---|
void |
computeMomentum() |
de.jungblut.math.DoubleVector |
minimize(de.jungblut.math.DoubleVector start,
java.util.function.Supplier<java.util.stream.Stream<de.jungblut.online.ml.FeatureOutcomePair>> streamSupplier,
StochasticCostFunction costFunction,
int numPasses,
boolean verbose)
Minimizes the given stochastic cost function on the supplied streams for
the given amount of passes over the data.
|
void |
setIterationCallback(IterationFinishedCallback iterationCallback) |
void |
setPassCallback(PassFinishedCallback passCallback) |
void |
setValidationCallback(ValidationFinishedCallback validationCallback) |
CostWeightTuple |
updateWeights(de.jungblut.math.minimize.CostGradientTuple observed) |
public de.jungblut.math.DoubleVector minimize(de.jungblut.math.DoubleVector start,
java.util.function.Supplier<java.util.stream.Stream<de.jungblut.online.ml.FeatureOutcomePair>> streamSupplier,
StochasticCostFunction costFunction,
int numPasses,
boolean verbose)
StochasticMinimizerminimize in interface StochasticMinimizerstart - the start parameters.streamSupplier - the supplier for the data training streams.costFunction - the cost function to minimize.numPasses - the number of passes over the streams.verbose - true if progress should be printed to the log.public void computeMomentum()
public CostWeightTuple updateWeights(de.jungblut.math.minimize.CostGradientTuple observed)
public void setIterationCallback(IterationFinishedCallback iterationCallback)
public void setValidationCallback(ValidationFinishedCallback validationCallback)
public void setPassCallback(PassFinishedCallback passCallback)
Copyright © 2015. All rights reserved.