ml.shifu.guagua.example.lnr

Class LinearRegressionMaster

java.lang.Object

ml.shifu.guagua.example.lnr.LinearRegressionMaster

All Implemented Interfaces:

ml.shifu.guagua.master.MasterComputable<LinearRegressionParams,LinearRegressionParams>

public class LinearRegressionMaster
extends Object
implements ml.shifu.guagua.master.MasterComputable<LinearRegressionParams,LinearRegressionParams>

LinearRegressionMaster defines logic to update global linear regression model.

At first iteration, master builds a random model then send to all workers to start computing. This is to make all workers use the same model at the starting time.

At other iterations, master works:

• 1. Accumulate all gradients from workers.
• 2. Update global models by using accumulated gradients.
• 3. Send new global model to workers by returning model parameters.

Constructor Summary

Constructors

Constructor and Description
LinearRegressionMaster()

Method Summary

Methods

Modifier and Type	Method and Description
LinearRegressionParams	compute(ml.shifu.guagua.master.MasterContext<LinearRegressionParams,LinearRegressionParams> context)

Methods inherited from class java.lang.Object

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

Constructor Detail

LinearRegressionMaster

public LinearRegressionMaster()

Method Detail

compute

public LinearRegressionParams compute(ml.shifu.guagua.master.MasterContext<LinearRegressionParams,LinearRegressionParams> context)

Specified by:

compute in interface ml.shifu.guagua.master.MasterComputable<LinearRegressionParams,LinearRegressionParams>