reagent.ope.datasets package
Submodules
reagent.ope.datasets.logged_dataset module
- class reagent.ope.datasets.logged_dataset.BanditsDataset
Bases:
abc.ABC
Base class for logged, aka behavior, dataset
- abstract property actions: torch.Tensor
- Returns
all actions in the dataset as numpy array
- abstract property features: torch.Tensor
- Returns
all features in the dataset as numpy array
- abstract property num_actions: int
- Returns
number of total possible actions
- abstract property num_features: int
- Returns
number of features
- abstract property rewards: torch.Tensor
- Returns
all rewards in the dataset as numpy array