org.deeplearning4j.spark.stats.ExampleCountEventStats Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dl4j-spark_2.11 Show documentation

There is a newer version: 1.0.0-beta_spark_2

package org.deeplearning4j.spark.stats;

import lombok.Getter;

/**
 * Event stats implementation with number of examples
 *
 * @author Alex Black
 */
public class ExampleCountEventStats extends BaseEventStats {

    @Getter private final int totalExampleCount;

    public ExampleCountEventStats(long startTime, long durationMs, int totalExampleCount) {
        super(startTime, durationMs);
        this.totalExampleCount = totalExampleCount;
    }

    public ExampleCountEventStats(String machineId, String jvmId, long threadId, long startTime, long durationMs, int totalExampleCount){
        super(machineId, jvmId, threadId, startTime, durationMs);
        this.totalExampleCount = totalExampleCount;
    }

    @Override
    public String asString(String delimiter){
        return super.asString(delimiter) + delimiter + totalExampleCount;
    }

    @Override
    public String getStringHeader(String delimiter){
        return super.getStringHeader(delimiter) + delimiter + "totalExampleCount";
    }
}