cosmo balance dataset hdr

function [balanced_ds, idxs, classes] = cosmo_balance_dataset(ds, varargin)