From 73601b555b422c70fe9c1241f8e66d11e89c452e Mon Sep 17 00:00:00 2001 From: VincentAuriau Date: Thu, 14 Mar 2024 15:01:17 +0100 Subject: [PATCH] ENH: small modifs --- choice_learn/datasets/data/__init__.py | 2 +- choice_learn/models/rumnet.py | 11 +++++++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/choice_learn/datasets/data/__init__.py b/choice_learn/datasets/data/__init__.py index d20a2dd6..c24dcc41 100644 --- a/choice_learn/datasets/data/__init__.py +++ b/choice_learn/datasets/data/__init__.py @@ -1 +1 @@ -"""Directory to store datasets as .csv.gz files.""" +"""Directory to store datasets as zipped .csv files.""" diff --git a/choice_learn/models/rumnet.py b/choice_learn/models/rumnet.py index a61eb9a8..3efd36ec 100644 --- a/choice_learn/models/rumnet.py +++ b/choice_learn/models/rumnet.py @@ -443,6 +443,17 @@ class PaperRUMnet(ChoiceModel): Representing Random Utility Choice Models with Neural Networks from Ali Aouad and Antoine Désir https://arxiv.org/abs/2207.12877 + --- Attention: --- + Note that the model uses two type of features that are treated differently: + - customer features + - product features + >>> In this implementation, please make sure that the features are correctly formatted: + - customer features: (n_contexts, n_features) are given as 'contexts_features' in the + ChoiceDataset used to fit the model + - product features: (n_contexts, n_items, n_features) are given as 'contexts_items_features' + in the ChoiceDataset used to fit the model + --- + Inherits from base_model.ChoiceModel TODO: Verify that all parameters are implemented. """