org.apache.ignite.ml.optimization.updatecalculators

Class SimpleGDParameterUpdate

java.lang.Object

org.apache.ignite.ml.optimization.updatecalculators.SimpleGDParameterUpdate

All Implemented Interfaces:

Serializable

public class SimpleGDParameterUpdate
extends Object
implements Serializable

Parameters for SimpleGDUpdateCalculator.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
SimpleGDParameterUpdate(int paramsCnt) Construct instance of this class.
SimpleGDParameterUpdate(Vector gradient) Construct instance of this class.

Method Summary

Modifier and Type	Method and Description
static SimpleGDParameterUpdate	avg(List<SimpleGDParameterUpdate> updates) Method used to get total update of all parallel trainings.
Vector	gradient() Get gradient.
static SimpleGDParameterUpdate	sumLocal(List<SimpleGDParameterUpdate> updates) Method used to sum updates inside of one of parallel trainings.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SimpleGDParameterUpdate

public SimpleGDParameterUpdate(int paramsCnt)

Construct instance of this class.

Parameters:

paramsCnt - Count of parameters.

SimpleGDParameterUpdate

public SimpleGDParameterUpdate(Vector gradient)

Construct instance of this class.

Parameters:

gradient - Gradient.

Method Detail

gradient

public Vector gradient()

Get gradient.

Returns:

Get gradient.

sumLocal

public static SimpleGDParameterUpdate sumLocal(List<SimpleGDParameterUpdate> updates)

Method used to sum updates inside of one of parallel trainings.

Parameters:

updates - Updates.

Returns:

Sum of SimpleGDParameterUpdate.

avg

public static SimpleGDParameterUpdate avg(List<SimpleGDParameterUpdate> updates)

Method used to get total update of all parallel trainings.

Parameters:

updates - Updates.

Returns:

Avg of SimpleGDParameterUpdate.