"""
Image Grid
===========
"""
import mosaic as ms
image_dir, data = ms.load_dataset('rothko')
with ms.data_context(image_col='image_path', image_dir=image_dir):
ms.image_grid(data=data, sort_by=ms.SATURATION, image_size=50)
"""
Scatter Grid
=============
"""
import mosaic as ms
image_dir, data = ms.load_dataset('mnist')
with ms.data_context(image_col='image_path', image_dir=image_dir):
ms.scatter_grid(x='x', y='y', data=data, hue='label', n_jobs=-1)
"""
Image BarPlot
=============
"""
import mosaic as ms
images, labels = ms.load_dataset('cifar10')
images = images[:1000]
labels = labels[:1000]
ms.image_barplot(y=labels, images=images, bar_height=50)