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.String
getLabel(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
-
-