org.deeplearning4j.spark.impl.common.DoubleAccumulator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dl4j-spark Show documentation

The newest version!

package org.deeplearning4j.spark.impl.common;

import org.apache.spark.Accumulator;
import org.apache.spark.AccumulatorParam;
import org.apache.spark.SparkContext;

/**
 * Created by nyghtowl on 1/14/16.
 */
public class DoubleAccumulator implements AccumulatorParam {

    @Override
    public Double addAccumulator(Double num, Double num1) {
        num += num1;
        return num;
        }

    @Override
    public Double addInPlace(Double num, Double num1) {
        num += num1;
        return num;
    }

    @Override
    public Double zero(Double num) {
            return num;
        }

    public static Accumulator create(SparkContext sc, String name) {
        return sc.accumulator(Double.POSITIVE_INFINITY, name, new DoubleAccumulator());
    }
}