Interface Labels

    • Method Detail

      • getLabel

        String getLabel​(int n)
        Returns the description of the nth class from the classes of a dataset.
        Parameters:
        n -
        Returns:
        label description
      • decodePredictions

        List<List<ClassPrediction>> decodePredictions​(INDArray predictions,
                                                      int n)
        Given predictions from the trained model this method will return a list of the top n matches and the respective probabilities.
        Parameters:
        predictions - raw
        Returns:
        decoded predictions