nlp/com.londogard.nlp.meachinelearning.optimizer/GradientDescent

GradientDescent

public final class GradientDescent

A simple implementation of Gradient Descent optimization

Constructors

GradientDescent

public GradientDescent GradientDescent( Integer maxIterations, Float stepSize, Float eps)

Functions

public final Float getEps()

public final Integer getI()

public final NDArray<Float, D1> getLosses()

getMaxIterations

public final Integer getMaxIterations()

public final Float getStepSize()

public final Pair<NDArray<Float, D2>, NDArray<Float, D1>> optimize( LogisticLoss loss, NDArray<Float, D2> initWeights, MultiArray<Float, D2> X, NDArray<Float, D2> y)

public final Unit setI(Integer i)

Properties

private final Float eps

private Integer i

private final NDArray<Float, D1> losses

private final Integer maxIterations

private final Float stepSize