org.deeplearning4j.zoo.util.ClassPrediction Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deeplearning4j-zoo Show documentation

There is a newer version: 1.0.0-M2.1

package org.deeplearning4j.zoo.util;

import lombok.AllArgsConstructor;
import lombok.Data;

/**
 * ClassPrediction: a prediction for classification, used with a {@link Labels} class.
 * Holds class number, label description, and the prediction probability.
 *
 * @author saudet
 */
@AllArgsConstructor
@Data
public class ClassPrediction {

    private int number;
    private String label;
    private double probability;

    @Override
    public String toString() {
        return "ClassPrediction(number=" + number + ",label=" + label + ",probability=" + probability + ")";
    }
}