ml.shifu.guagua.example.lnr

Class LinearRegressionMaster

java.lang.Object

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

ml.shifu.guagua.example.lnr.LinearRegressionMaster

All Implemented Interfaces:

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

public class LinearRegressionMaster
extends ml.shifu.guagua.master.AbstractMasterComputable<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.

Field Summary

Fields inherited from class ml.shifu.guagua.master.AbstractMasterComputable

isInitialized

Constructor Summary

Constructors

Constructor and Description
LinearRegressionMaster()

Method Summary

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

Methods inherited from class ml.shifu.guagua.master.AbstractMasterComputable

compute

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

init

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

Specified by:

init in class ml.shifu.guagua.master.AbstractMasterComputable<LinearRegressionParams,LinearRegressionParams>

doCompute

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

Specified by:

doCompute in class ml.shifu.guagua.master.AbstractMasterComputable<LinearRegressionParams,LinearRegressionParams>