print_top_classes

txv.utils.print_top_classes(predictions, ntop=10, **kwargs)

Parameters:

• predictions (torch.Tensor) – The output of the model.

• ntop (int, optional) – The number of top classes to print. The default is 10.