ml.shifu.guagua.example.lr

Class LogisticRegressionMaster

java.lang.Object

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

ml.shifu.guagua.example.lr.LogisticRegressionMaster

All Implemented Interfaces:

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

public class LogisticRegressionMaster
extends ml.shifu.guagua.master.AbstractMasterComputable<LogisticRegressionParams,LogisticRegressionParams>

LogisticRegressionMaster defines logic to update global logistic 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
LogisticRegressionMaster()

Method Summary

Modifier and Type	Method and Description
LogisticRegressionParams	doCompute(ml.shifu.guagua.master.MasterContext<LogisticRegressionParams,LogisticRegressionParams> context)
void	init(ml.shifu.guagua.master.MasterContext<LogisticRegressionParams,LogisticRegressionParams> 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

LogisticRegressionMaster

public LogisticRegressionMaster()

Method Detail

init

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

Specified by:

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

doCompute

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

Specified by:

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