public class FederatedPSControlThread extends PSWorker implements Callable<Void>
Constructor and Description |
---|
FederatedPSControlThread(int workerID,
String updFunc,
Statement.PSFrequency freq,
Statement.PSRuntimeBalancing runtimeBalancing,
boolean weighting,
int epochs,
long batchSize,
int numBatchesPerGlobalEpoch,
ExecutionContext ec,
ParamServer ps) |
Modifier and Type | Method and Description |
---|---|
Void |
call()
Entry point of the functionality
|
String |
getWorkerName() |
void |
setup(double weightingFactor)
Sets up the federated worker and control thread
|
void |
teardown()
cleans up the execution context of the federated worker
|
getFeatures, getLabels, setFeatures, setLabels
public FederatedPSControlThread(int workerID, String updFunc, Statement.PSFrequency freq, Statement.PSRuntimeBalancing runtimeBalancing, boolean weighting, int epochs, long batchSize, int numBatchesPerGlobalEpoch, ExecutionContext ec, ParamServer ps)
public void setup(double weightingFactor)
weightingFactor
- Gradients from this worker will be multiplied by this factor if weighting is enabledpublic void teardown()
public String getWorkerName()
getWorkerName
in class PSWorker
Copyright © 2021 The Apache Software Foundation. All rights reserved.