def __init__(self, n_features):
BaseBandit.__init__(self)
self.n_features = n_features
def __init__(self, n_train, n_test, n_splits, *args, **kwargs):
BaseBandit.__init__(self, *args, **kwargs)
self.n_train = n_train
self.n_test = n_test
self.n_splits = n_splits