Class SingleLabelClassification

  • All Implemented Interfaces:
    org.api4.java.ai.ml.classification.singlelabel.evaluation.ISingleLabelClassification, org.api4.java.ai.ml.core.evaluation.IPrediction

    public class SingleLabelClassification
    extends Prediction
    implements org.api4.java.ai.ml.classification.singlelabel.evaluation.ISingleLabelClassification
    • Constructor Detail

      • SingleLabelClassification

        public SingleLabelClassification​(int predicted)
      • SingleLabelClassification

        public SingleLabelClassification​(java.util.Map<java.lang.Integer,​java.lang.Double> labelProbabilities)
    • Method Detail

      • getIntPrediction

        public int getIntPrediction()
        Specified by:
        getIntPrediction in interface org.api4.java.ai.ml.classification.singlelabel.evaluation.ISingleLabelClassification
      • getPrediction

        public java.lang.Integer getPrediction()
        Specified by:
        getPrediction in interface org.api4.java.ai.ml.core.evaluation.IPrediction
        Specified by:
        getPrediction in interface org.api4.java.ai.ml.classification.singlelabel.evaluation.ISingleLabelClassification
        Overrides:
        getPrediction in class Prediction
      • getLabelWithHighestProbability

        public java.lang.Integer getLabelWithHighestProbability()
        Specified by:
        getLabelWithHighestProbability in interface org.api4.java.ai.ml.core.evaluation.IPrediction
        Specified by:
        getLabelWithHighestProbability in interface org.api4.java.ai.ml.classification.singlelabel.evaluation.ISingleLabelClassification
        Overrides:
        getLabelWithHighestProbability in class Prediction
      • getClassDistribution

        public java.util.Map<java.lang.Integer,​java.lang.Double> getClassDistribution()
        Specified by:
        getClassDistribution in interface org.api4.java.ai.ml.core.evaluation.IPrediction
        Specified by:
        getClassDistribution in interface org.api4.java.ai.ml.classification.singlelabel.evaluation.ISingleLabelClassification
        Overrides:
        getClassDistribution in class Prediction
      • getProbabilityOfLabel

        public double getProbabilityOfLabel​(int label)
        Specified by:
        getProbabilityOfLabel in interface org.api4.java.ai.ml.classification.singlelabel.evaluation.ISingleLabelClassification
      • getClassConfidence

        public java.util.Map<java.lang.Integer,​java.lang.Double> getClassConfidence()
        Specified by:
        getClassConfidence in interface org.api4.java.ai.ml.core.evaluation.IPrediction
        Specified by:
        getClassConfidence in interface org.api4.java.ai.ml.classification.singlelabel.evaluation.ISingleLabelClassification
        Overrides:
        getClassConfidence in class Prediction