seldonian.dataset

Build and load datasets for running Seldonian algorithms

Functions

load_RL_metadata(metadata_filename, ...)

Load RL metadata from JSON file into a dictionary

load_custom_metadata(filename)

Load custom regime metadata from JSON file into a dictionary

load_supervised_metadata(filename)

Load metadata from JSON file into a dictionary

Classes

CustomDataSet(data, sensitive_attrs, ...)

CustomMetaData(all_col_names[, ...])

DataSet(num_datapoints, meta, regime, **kwargs)

DataSetLoader(regime, **kwargs)

Episode(observations, actions, rewards, ...)

MetaData(regime, sub_regime, all_col_names)

RLDataSet(episodes, meta[, sensitive_attrs])

RLMetaData(all_col_names[, sensitive_col_names])

SupervisedDataSet(features, labels, ...)

SupervisedMetaData(sub_regime, ...[, ...])