public class SparkPSWorker extends LocalPSWorker implements org.apache.spark.api.java.function.VoidFunction<scala.Tuple2<Integer,scala.Tuple2<MatrixBlock,MatrixBlock>>>
Constructor and Description |
---|
SparkPSWorker(String updFunc,
String aggFunc,
Statement.PSFrequency freq,
int epochs,
long batchSize,
String program,
HashMap<String,byte[]> clsMap,
org.apache.spark.SparkConf conf,
int port,
org.apache.spark.util.LongAccumulator aSetup,
org.apache.spark.util.LongAccumulator aWorker,
org.apache.spark.util.LongAccumulator aUpdate,
org.apache.spark.util.LongAccumulator aIndex,
org.apache.spark.util.LongAccumulator aGrad,
org.apache.spark.util.LongAccumulator aRPC,
org.apache.spark.util.LongAccumulator aBatches,
org.apache.spark.util.LongAccumulator aEpochs) |
Modifier and Type | Method and Description |
---|---|
void |
call(scala.Tuple2<Integer,scala.Tuple2<MatrixBlock,MatrixBlock>> input) |
String |
getWorkerName() |
call
getFeatures, getLabels, setFeatures, setLabels
public SparkPSWorker(String updFunc, String aggFunc, Statement.PSFrequency freq, int epochs, long batchSize, String program, HashMap<String,byte[]> clsMap, org.apache.spark.SparkConf conf, int port, org.apache.spark.util.LongAccumulator aSetup, org.apache.spark.util.LongAccumulator aWorker, org.apache.spark.util.LongAccumulator aUpdate, org.apache.spark.util.LongAccumulator aIndex, org.apache.spark.util.LongAccumulator aGrad, org.apache.spark.util.LongAccumulator aRPC, org.apache.spark.util.LongAccumulator aBatches, org.apache.spark.util.LongAccumulator aEpochs)
public String getWorkerName()
getWorkerName
in class LocalPSWorker
public void call(scala.Tuple2<Integer,scala.Tuple2<MatrixBlock,MatrixBlock>> input) throws Exception
call
in interface org.apache.spark.api.java.function.VoidFunction<scala.Tuple2<Integer,scala.Tuple2<MatrixBlock,MatrixBlock>>>
Exception
Copyright © 2020 The Apache Software Foundation. All rights reserved.