Add in data exploration support

[jimmyjbling]

• [x] umap
• [x] diversity calculation
• [ ] label stats (balancing/histogram)