@Sayed_Nadim  I want to display the results of the classification (prediction of the code) in image, how?
I tested by this code which gave by @ptrblck  but still displays this image
print('Validation accuracy test: {:.4f}%'.format(float(accuracy_score(val_y, predictions)) * 100))
conf_matrix  = confusion_matrix(val_y, predictions)
print(conf_matrix)
predictions = torch.randn(1, 1, 224, 224)
plt.imshow(torchvision.utils.make_grid(predictions).permute(1, 2, 0))