--- title: MovieLens Dataset keywords: fastai sidebar: home_sidebar summary: "Implementation of MovieLens datasets." description: "Implementation of MovieLens datasets." nb_path: "nbs/datasets/datasets.movielens.ipynb" ---
class Args:
def __init__(self):
self.data_dir = '/content/data'
self.min_rating = 4
self.num_negative_samples = 99
self.min_uc = 5
self.min_sc = 5
self.val_p = 0.2
self.test_p = 0.2
self.seed = 42
self.split_type = 'stratified'
args = Args()
ds = ML1mDataModule(**args.__dict__)
ds.prepare_data()
!tree -h --du -C "{args.data_dir}"