src package

Subpackages

• src.datamodules package
  • Subpackages
    • src.datamodules.components package
      • Submodules
      • src.datamodules.components.utils module
      • Module contents
    • src.datamodules.mind package
      • Submodules
      • src.datamodules.mind.datamodule_BERT module
      • src.datamodules.mind.datamodule_Base module
      • src.datamodules.mind.datamodule_CollaborativeFiltering module
      • src.datamodules.mind.datamodule_MKR module
      • src.datamodules.mind.datamodule_NAML module
      • src.datamodules.mind.datamodule_RippleNet module
      • src.datamodules.mind.dataset module
      • src.datamodules.mind.download module
      • src.datamodules.mind.parse module
      • src.datamodules.mind.preprocessing module
      • Module contents
  • Module contents
• src.models package
  • Subpackages
    • src.models.BERT package
      • Module contents
    • src.models.DKN package
      • Submodules
      • src.models.DKN.KCNN module
      • src.models.DKN.attention module
      • Module contents
    • src.models.LSTUR package
      • Submodules
      • src.models.LSTUR.news_encoder module
      • src.models.LSTUR.user_encoder module
      • Module contents
    • src.models.MKR package
      • Submodules
      • src.models.MKR.kGDataset module
      • src.models.MKR.layers module
      • Module contents
    • src.models.collaborativeFiltering package
      • Module contents
    • src.models.components package
      • Subpackages
      • Submodules
      • src.models.components.metrics module
      • src.models.components.simple_dense_net module
      • src.models.components.utils module
      • Module contents
    • src.models.naml package
      • Submodules
      • src.models.naml.news_encoder module
      • src.models.naml.user_encoder module
      • Module contents
    • src.models.nrms package
      • Submodules
      • src.models.nrms.news_encoder module
      • src.models.nrms.user_encoder module
      • Module contents
    • src.models.rippleNet package
      • Module contents
  • Module contents
• src.utils package
  • Module contents
    • extras()
    • finish()
    • get_logger()
    • log_hyperparameters()
    • print_config()
• src.vendor package
  • Module contents

Submodules

src.testing_pipeline module

src.testing_pipeline.test(config: DictConfig) → None

Contains minimal example of the testing pipeline. Evaluates given checkpoint on a testset.

Parameters:

config (DictConfig) – Configuration composed by Hydra.

Returns:

None

src.training_pipeline module

src.training_pipeline.train(config: DictConfig) → Optional[float]

Contains the training pipeline. Can additionally evaluate model on a testset, using best weights achieved during training.

Parameters:

config (DictConfig) – Configuration composed by Hydra.

Returns:

Metric score for hyperparameter optimization.

Return type:

Optional[float]

Module contents