Learning-based density-equalizing map