experiments.experiment_utils.prep_custom_data

prep_custom_data(trial_dataset, n_points, include_sensitive_attrs=False)

Utility function for preparing data and sensitive attributes for the custom regime for a given trial with n_points (given data frac)

Parameters:
  • trial_dataset – The Seldonian dataset object containing trial data

  • n_points (int) – Number of points in this trial

  • include_sensitive_attrs (bool) – Whether to prep and return sensitive attributes as well.