I didn't think this all the way through and ran UMAP on each of these groups separately instead of at the same time.