weka.dl4j.inference

Class TopNPredictions

java.lang.Object

weka.dl4j.inference.TopNPredictions

public class TopNPredictions
extends java.lang.Object

Holds an arbitrary number of predictions, ordered by class probability.

Constructor Summary

Constructors

Constructor and Description
TopNPredictions() Initialize.
TopNPredictions(java.lang.String imageName, java.lang.String modelName) Initialize the top N predictions.

Method Summary

Modifier and Type	Method and Description
int	getN()
Prediction	getPrediction(int i) Get the prediction at index i.
Prediction	getTopPrediction()
Prediction[]	getTopPredictions()
void	process(org.nd4j.linalg.api.ndarray.INDArray predictions, java.lang.String[] classes) Main entrypoint, decodes predictions into a TopNPredictions object.
void	setN(int n)
java.lang.String	toSummaryString() Return a summary string of the stored predictions.
java.lang.String	toSummaryString(java.lang.String imageName, java.lang.String zooModelName) Return a summary string of the stored predictions, headed with the given image and model name.

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TopNPredictions

public TopNPredictions()

Initialize.

TopNPredictions

public TopNPredictions(java.lang.String imageName,
                       java.lang.String modelName)

Initialize the top N predictions.

Parameters:

imageName - Name of image

modelName - Name of model

Method Detail

process

public void process(org.nd4j.linalg.api.ndarray.INDArray predictions,
                    java.lang.String[] classes)
             throws java.lang.Exception

Main entrypoint, decodes predictions into a TopNPredictions object.

Parameters:

predictions - Raw model predictions

classes - Class map

Throws:

java.lang.Exception - image exception

toSummaryString

public java.lang.String toSummaryString()

Return a summary string of the stored predictions.

Returns:

Results in table format

toSummaryString

public java.lang.String toSummaryString(java.lang.String imageName,
                                        java.lang.String zooModelName)

Return a summary string of the stored predictions, headed with the given image and model name.

Parameters:

imageName - Image used for prediction

zooModelName - Model used for prediction

Returns:

Results in table format

getN

public int getN()

setN

public void setN(int n)

getPrediction

public Prediction getPrediction(int i)

Get the prediction at index i.

Parameters:

i - Index of prediction to get.

Returns:

Prediction.

getTopPrediction

public Prediction getTopPrediction()

getTopPredictions

public Prediction[] getTopPredictions()