Interface IClassificationPredictor

  • All Superinterfaces:
    IPredictorInfo

    public interface IClassificationPredictor
    extends IPredictorInfo
    Interface for multi class and binary classification models.
    • Method Detail

      • getClassLabels

        String[] getClassLabels()
        Model specific class labels that will be predicted.

        NOTE:

        These labels will be used as a key set for outputs of score(Iterable) and score(Map).

        Returns:
        array of class labels.
      • score

        Map<String,​Double> score​(Map<String,​?> row)
        Scores a single row.
        Parameters:
        row - row with feature values.
        Returns:
        class label to its probability.
      • score

        Map<String,​Double> score​(Iterable<?> row)
        Scores a single row.

        NOTE:

        Here Iterable<?> row is the feature values in the same order as they goes in IPredictorInfo.getFeatures().

        Parameters:
        row - row with feature values in right order.
        Returns:
        class label to its probability.