ENH: better API for passing weighted_n_samples