Package org.deeplearning4j.zoo.util
Interface Labels
-
- All Known Implementing Classes:
BaseLabels,COCOLabels,DarknetLabels,ImageNetLabels,VOCLabels
public interface Labels
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description 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.StringgetLabel(int n)Returns the description of the nth class from the classes of a dataset.
-
-
-
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
-
-