org.deeplearning4j.ui.weights.beans.CompactModelAndGradient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of deeplearning4j-ui-model Show documentation
There is a newer version: 1.0.0-M2.1
Show newest version
package org.deeplearning4j.ui.weights.beans;


import java.io.Serializable;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Slightly modified version of ModelAndGradient, with binned params/gradients, suitable for fast network transfers for HistogramIterationListener
 *
 * @author Adam Gibson
 */

public class CompactModelAndGradient implements Serializable {
    private long lastUpdateTime = -1L;
    private Map parameters;
    private Map gradients;
    private double score;
    private List scores = new ArrayList<>();
    private List>> updateMagnitudes = new ArrayList<>();
    private List>> paramMagnitudes = new ArrayList<>();
    private List layerNames = new ArrayList<>();
    private String path;


    public CompactModelAndGradient() {
        parameters = new HashMap<>();
        gradients = new HashMap<>();
    }


    public void setLastUpdateTime(long lastUpdateTime) {
        this.lastUpdateTime = lastUpdateTime;
    }

    public long getLastUpdateTime() {
        return lastUpdateTime;
    }

    public double getScore() {
        return score;
    }

    public void setScore(double score) {
        this.score = score;
    }


    public Map getParameters() {
        return parameters;
    }

    public void setParameters(Map parameters) {
        this.parameters = parameters;
    }


    public Map getGradients() {
        return gradients;
    }

    public void setGradients(Map gradients) {
        this.gradients = gradients;
    }

    public void setScores(List scores) {
        this.scores = scores;
    }

    public void setPath(String path) {
        this.path = path;
    }

    public String getPath() {
        return path;
    }

    public List getScores() {
        return scores;
    }

    public void setUpdateMagnitudes(List>> updateMagnitudes) {
        this.updateMagnitudes = updateMagnitudes;
    }

    public List>> getUpdateMagnitudes() {
        return updateMagnitudes;
    }

    public void setParamMagnitudes(List>> paramMagnitudes) {
        this.paramMagnitudes = paramMagnitudes;
    }

    public List>> getParamMagnitudes() {
        return paramMagnitudes;
    }

    public void setLayerNames(List layerNames) {
        this.layerNames = layerNames;
    }

    public List getLayerNames() {
        return layerNames;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o)
            return true;
        if (o == null || getClass() != o.getClass())
            return false;

        CompactModelAndGradient that = (CompactModelAndGradient) o;

        if (Double.compare(that.score, score) != 0)
            return false;
        if (parameters != null ? !parameters.equals(that.parameters) : that.parameters != null)
            return false;
        return !(gradients != null ? !gradients.equals(that.gradients) : that.gradients != null);
    }

    @Override
    public int hashCode() {
        int result;
        long temp;
        result = parameters != null ? parameters.hashCode() : 0;
        result = 31 * result + (gradients != null ? gradients.hashCode() : 0);
        temp = Double.doubleToLongBits(score);
        result = 31 * result + (int) (temp ^ (temp >>> 32));
        return result;
    }
}