thank you for your reply @ptrblck
But when I tested this :
conf_matrix  = confusion_matrix(val_y, predictions)

        

print(conf_matrix)

plt.imshow(torchvision.utils.make_grid((predictions, nrow: int = 8, padding: int = 2, normalize: bool = False, value_range: Optional[Tuple[int, int]] = None, scale_each: bool = False, pad_value: int = 0, **kwargs))

Error :
File “”, line 403
plt.imshow(torchvision.utils.make_grid((predictions, nrow: int = 8, padding: int = 2, normalize: bool = False, value_range: Optional[Tuple[int, int]] = None, scale_each: bool = False, pad_value: int = 0, **kwargs))
^
SyntaxError: invalid syntax