--- title: Weeplaces keywords: fastai sidebar: home_sidebar summary: "Implementation of Weeplaces dataset." description: "Implementation of Weeplaces dataset." nb_path: "nbs/datasets/datasets.weeplaces.ipynb" ---
root = './data'
# Define train/val/test datasets on user interactions.
train_dataset = WeeplacesDataset(root, is_group=False, datatype='train') # train dataset for user-item interactions.
n_users, n_items = train_dataset.n_users, train_dataset.n_items
val_dataset = WeeplacesDataset(root, is_group=False, datatype='val', n_items=n_items)
test_dataset = WeeplacesDataset(root, is_group=False, datatype='test', n_items=n_items)
# Define train/val/test datasets on group and user interactions.
train_group_dataset = WeeplacesDataset(root, is_group=True, datatype='train', negs_per_group=5, n_items=n_items)
padding_idx = train_group_dataset.padding_idx
val_group_dataset = WeeplacesDataset(root, is_group=True, datatype='val', n_items=n_items, padding_idx=padding_idx)
test_group_dataset = WeeplacesDataset(root, is_group=True, datatype='test', n_items=n_items, padding_idx=padding_idx)
!tree --du -h -C ./data