deepr.examples.movielens.readers package

Submodules

deepr.examples.movielens.readers.csv module

CSV Reader for MovieLens.

class deepr.examples.movielens.readers.csv.TestCSVReader(path_csv_tr, path_csv_te, vocab_size)[source]

Bases: Reader

Reader of MovieLens test CSV files of the Multi-VAE paper.

See https://github.com/dawenl/vae_cf

as_dataset()[source]

Build a tf.data.Dataset

class deepr.examples.movielens.readers.csv.TrainCSVReader(path_csv, vocab_size, target_ratio=None, bucket_size=8192, shuffle=True, seed=42, take_ratio=None)[source]

Bases: Reader

Reader of MovieLens CSV files of the Multi-VAE paper.

See https://github.com/dawenl/vae_cf

as_dataset()[source]

Build a tf.data.Dataset

Module contents