deepr.examples.movielens package

Subpackages

• deepr.examples.movielens.jobs package
  • Submodules
  • deepr.examples.movielens.jobs.build_records module
    • BuildRecords
      • BuildRecords.min_length
      • BuildRecords.min_rating
      • BuildRecords.num_negatives
      • BuildRecords.path_eval
      • BuildRecords.path_mapping
      • BuildRecords.path_ratings
      • BuildRecords.path_test
      • BuildRecords.path_train
      • BuildRecords.run()
      • BuildRecords.seed
      • BuildRecords.shuffle_timelines
      • BuildRecords.size_eval
      • BuildRecords.size_test
      • BuildRecords.target_ratio
    • get_timelines()
    • records_generator()
    • write_records()
  • deepr.examples.movielens.jobs.evaluate module
    • Evaluate
      • Evaluate.k
      • Evaluate.num_queries
      • Evaluate.path_biases
      • Evaluate.path_embeddings
      • Evaluate.path_predictions
      • Evaluate.run()
      • Evaluate.use_mlflow
    • compute_metrics()
    • ndcg_score()
    • precision_recall_f1()
  • deepr.examples.movielens.jobs.init module
    • InitCheckpoint
      • InitCheckpoint.normalize
      • InitCheckpoint.normalize_counts
      • InitCheckpoint.path_counts
      • InitCheckpoint.path_embeddings
      • InitCheckpoint.path_init_ckpt
      • InitCheckpoint.run()
      • InitCheckpoint.use_log_counts
  • deepr.examples.movielens.jobs.predict module
    • Predict
      • Predict.input_fn
      • Predict.path_predictions
      • Predict.path_saved_model
      • Predict.prepro_fn
      • Predict.run()
  • deepr.examples.movielens.jobs.svd module
    • SVD
      • SVD.dim
      • SVD.min_count
      • SVD.path_counts
      • SVD.path_csv
      • SVD.path_embeddings
      • SVD.run()
      • SVD.vocab_size
    • compute_coocurrence()
    • compute_pmi()
  • Module contents
• deepr.examples.movielens.layers package
  • Submodules
  • deepr.examples.movielens.layers.average module
    • AddBias
      • AddBias.forward()
    • AverageModel()
    • Logits
      • Logits.forward()
    • Projection
      • Projection.forward()
    • UserEmbedding
      • UserEmbedding.forward()
  • deepr.examples.movielens.layers.bpr module
    • BPRLoss()
  • deepr.examples.movielens.layers.losses module
    • L2Loss
      • L2Loss.forward()
    • Loss()
    • VAELoss()
  • deepr.examples.movielens.layers.multi module
    • MultiLogLikelihoodCSS()
  • deepr.examples.movielens.layers.ns module
    • NegativeSampling()
  • deepr.examples.movielens.layers.transformer module
    • Logits
      • Logits.forward()
    • TransformerModel()
  • deepr.examples.movielens.layers.vae module
    • AddBias
      • AddBias.forward()
    • Decode
      • Decode.forward()
    • Encode
      • Encode.forward()
    • GaussianNoise
      • GaussianNoise.forward()
    • Logits
      • Logits.forward()
    • Projection
      • Projection.forward()
    • VAEModel()
    • WeightedSum
      • WeightedSum.forward()
  • Module contents
• deepr.examples.movielens.macros package
  • Submodules
  • deepr.examples.movielens.macros.paths module
    • Paths
  • Module contents
• deepr.examples.movielens.metrics package
  • Submodules
  • deepr.examples.movielens.metrics.ndcg module
    • NDCGAtK
      • NDCGAtK.name
      • NDCGAtK.logits
      • NDCGAtK.targets
      • NDCGAtK.k
      • NDCGAtK.inputs
  • deepr.examples.movielens.metrics.recall module
    • RecallAtK
      • RecallAtK.name
      • RecallAtK.logits
      • RecallAtK.targets
      • RecallAtK.k
      • RecallAtK.inputs
  • Module contents
• deepr.examples.movielens.prepros package
  • Submodules
  • deepr.examples.movielens.prepros.csv module
    • CSVPrepro()
    • RandomNegatives
      • RandomNegatives.forward()
    • SequenceMask
      • SequenceMask.forward()
  • deepr.examples.movielens.prepros.record module
    • RecordPrepro()
    • SequenceMask
      • SequenceMask.forward()
  • Module contents
• deepr.examples.movielens.readers package
  • Submodules
  • deepr.examples.movielens.readers.csv module
    • TestCSVReader
      • TestCSVReader.as_dataset()
    • TrainCSVReader
      • TrainCSVReader.as_dataset()
  • Module contents
• deepr.examples.movielens.utils package
  • Submodules
  • deepr.examples.movielens.utils.fields module
    • INPUT_POSITIVES_ONE_HOT()
    • TARGET_POSITIVES_ONE_HOT()
  • Module contents

Module contents