gsbDBI / torch-choice

Choice modeling with PyTorch: logit model and nested logit model
MIT License
39 stars 8 forks source link

Allowing for `user-item` and `user-session-item` specific observables. #28

Closed TianyuDu closed 1 year ago

TianyuDu commented 1 year ago

The packge currently support the following four types of observables:

image

We are implementing new features so that the package supports useritem_obs and useritemsession_obs observables. As you would expect, they would have shapes of (num_users, num_items, num_variables) and (num_users, num_items, num_sessions, num_variables).

The following components need to be updated.