fidelity / mab2rec

[AAAI 2024] Mab2Rec: Multi-Armed Bandits Recommender

Home Page:https://fidelity.github.io/mab2rec/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

user_id not in data file.

ayush488 opened this issue · comments

Not able to understand why I am getting a ValueError: user_id not in data file
this is my statement:

reco_to_results, reco_to_metrics = benchmark(recommenders, 
                                             metrics=metrics,
                                             train_data=X_train, 
                                             test_data=X_test,
                                             user_features=df_users_X, 
                                             user_id_col= 'ID',
                                               item_id_col= 'MailerID',
                                               response_col = 'sales_net')   

I think you are not passing the user_id , item_id, response_col in the load_response data function. Please check.

Hey ayush488! You are correct, thanks for bringing this to our attention. We will push a fix as soon as possible.