src.datamodules package

Subpackages

• src.datamodules.components package
  • Submodules
  • src.datamodules.components.utils module
    • download_path()
    • maybe_download()
    • unzip_file()
  • Module contents
• src.datamodules.mind package
  • Submodules
  • src.datamodules.mind.datamodule_BERT module
    • MINDDataModuleBERT
      • MINDDataModuleBERT.news_dataframe()
      • MINDDataModuleBERT.prepare_data()
      • MINDDataModuleBERT.setup()
      • MINDDataModuleBERT.test_dataloader()
      • MINDDataModuleBERT.train_dataloader()
      • MINDDataModuleBERT.val_dataloader()
  • src.datamodules.mind.datamodule_Base module
    • MINDDataModule
      • MINDDataModule.prepare()
      • MINDDataModule.setup()
      • MINDDataModule.test_dataloader()
      • MINDDataModule.train_dataloader()
      • MINDDataModule.val_dataloader()
  • src.datamodules.mind.datamodule_CollaborativeFiltering module
    • MINDDataModuleCollaborativeFiltering
      • MINDDataModuleCollaborativeFiltering.prepare()
      • MINDDataModuleCollaborativeFiltering.prepare_data()
      • MINDDataModuleCollaborativeFiltering.setup()
      • MINDDataModuleCollaborativeFiltering.train_dataloader()
  • src.datamodules.mind.datamodule_MKR module
    • MINDDataModuleMKR
      • MINDDataModuleMKR.prepare()
      • MINDDataModuleMKR.prepare_data()
      • MINDDataModuleMKR.setup()
      • MINDDataModuleMKR.train_dataloader()
  • src.datamodules.mind.datamodule_NAML module
    • MINDDataModuleNAML
      • MINDDataModuleNAML.news_dataloader()
      • MINDDataModuleNAML.prepare()
      • MINDDataModuleNAML.setup()
      • MINDDataModuleNAML.test_dataloader()
      • MINDDataModuleNAML.user_dataloader()
      • MINDDataModuleNAML.val_dataloader()
  • src.datamodules.mind.datamodule_RippleNet module
    • MINDDataModuleRippleNet
      • MINDDataModuleRippleNet.prepare()
      • MINDDataModuleRippleNet.prepare_data()
      • MINDDataModuleRippleNet.setup()
      • MINDDataModuleRippleNet.train_dataloader()
  • src.datamodules.mind.dataset module
    • BaseDataset
    • BehaviorsBERTDataset
    • BehaviorsDataset
    • KGDataset
    • NewsBERTDataset
    • NewsDataset
      • NewsDataset.to()
    • RatingsDataset
    • UserDataset
  • src.datamodules.mind.download module
    • download_and_extract_glove()
    • download_and_extract_mind()
    • download_and_extract_wikidata_kg()
    • extract_mind()
    • generate_embeddings()
    • get_train_input()
    • get_user_history()
    • get_valid_input()
    • load_glove_matrix()
    • read_clickhistory()
    • read_news()
    • read_news_ids()
    • word_tokenize()
  • src.datamodules.mind.parse module
    • generate_word_embedding()
    • parse_behaviors()
    • parse_behaviors_bert()
    • parse_mind()
    • parse_mind_bert()
    • parse_news()
    • parse_news_bert()
    • transform_entity_embedding()
  • src.datamodules.mind.preprocessing module
    • create_knowledge_graph_file()
    • create_rating_file()
    • create_rating_file_collaborative()
    • prepare_numpy_data()
    • prepare_numpy_kg()
  • Module contents

Module contents