from dlutils import plot_image_batch_w_labels

# where `image_batch` is the batch of images to be plotted
# `label_batch` is a batch of true labels corresponding to each image in `image_batch`
# `predicted_labels` is a batch of labels predicted by our model for each image in `image_batch`

label_batch_strings = []
    for true_label, predicted_label in zip(label_batch, predicted_labels):
        label_batch_strings.append('True: {}, Predicted: {}'.format(true_label, predicted_label))
        
plot_image_batch_w_labels.plot_batch(image_batch, os.path.join(path, 'test.png'), label_batch=label_batch_strings)