--- title: Tmall dataset keywords: fastai sidebar: home_sidebar summary: "Tmall dataset." description: "Tmall dataset." nb_path: "nbs/datasets/tmall.ipynb" ---
{% raw %}
{% endraw %} {% raw %}
{% endraw %} {% raw %}

class TmallDataset[source]

TmallDataset(root, shuffle=False, n_node=40727, is_train=True) :: SessionGraphDataset

References

1. COTREC session-based recommender model training. https://t.ly/cXTH.
{% endraw %} {% raw %}
{% endraw %} {% raw %}
root = '/content/tmall'

train_data = TmallDataset(root=root, shuffle=True, is_train=True)
test_data = TmallDataset(root=root, shuffle=False, is_train=False)
{% endraw %}