Method Detail

• getLabel

  java.lang.String getLabel(int n)

  Returns the description of the nth class from the classes of a dataset.

  Parameters:

  n -

  Returns:

  label description

• decodePredictions

  java.util.List<java.util.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