function [balanced_ds, idxs, classes] = cosmo_balance_dataset(ds, varargin)
cosmo average samples hdr
cosmo balance partitions hdr