diff --git a/choice_learn/data/choice_dataset.py b/choice_learn/data/choice_dataset.py index 4bdee821..848bdc70 100644 --- a/choice_learn/data/choice_dataset.py +++ b/choice_learn/data/choice_dataset.py @@ -52,8 +52,8 @@ def __init__( over the different choices, default is None features_by_ids : list of (name, FeaturesStorage) List of Storage objects. Their name must correspond to a feature name - among shared_items, contexts, - contexts_items and their ids must match to those features values. Default is [] + among shared_features_by_choice or items_features_by_choice + and their ids must match to those features values. Default is [] shared_features_by_choice_names : tuple of (array_like, ) list of names of the contexts_features, default is None items_features_by_choice_names : tuple of (array_like, )