Skip to content

BanditDataset

Base class for bandit datasets.

Parameters

  • n_features

    Number of features in the dataset.

  • n_samples

    DefaultNone

    Number of samples in the dataset.

  • n_classes

    DefaultNone

    Number of classes in the dataset, only applies to classification datasets.

  • n_outputs

    DefaultNone

    Number of outputs the target is made of, only applies to multi-output datasets.

  • sparse

    DefaultFalse

    Whether the dataset is sparse or not.

Attributes

  • arms

    The list of arms that can be pulled.

  • desc

    Return the description from the docstring.

Methods

take

Iterate over the k samples.

Parameters

  • k'int'