src.models package§
Subpackages§
- src.models.BERT package
- src.models.DKN package
- Submodules
- src.models.DKN.KCNN module
- src.models.DKN.attention module
- Module contents
DKNDKN.configure_optimizers()DKN.eval_epoch_end()DKN.eval_step()DKN.forward()DKN.get_news_vector()DKN.get_prediction()DKN.get_user_vector()DKN.on_eval_start()DKN.on_test_start()DKN.on_train_start()DKN.on_validation_start()DKN.test_epoch_end()DKN.test_step()DKN.trainingDKN.training_step()DKN.validation_epoch_end()DKN.validation_step()
- src.models.LSTUR package
- Submodules
- src.models.LSTUR.news_encoder module
- src.models.LSTUR.user_encoder module
- Module contents
LSTURLSTUR.configure_optimizers()LSTUR.eval_epoch_end()LSTUR.eval_step()LSTUR.forward()LSTUR.get_news_vector()LSTUR.get_prediction()LSTUR.get_user_vector()LSTUR.on_eval_start()LSTUR.on_test_start()LSTUR.on_train_start()LSTUR.on_validation_start()LSTUR.test_epoch_end()LSTUR.test_step()LSTUR.trainingLSTUR.training_step()LSTUR.validation_epoch_end()LSTUR.validation_step()
- src.models.MKR package
- Submodules
- src.models.MKR.kGDataset module
- src.models.MKR.layers module
- Module contents
MKRMKR.configure_optimizers()MKR.eval_epoch_end()MKR.eval_step()MKR.eval_step_end()MKR.forward()MKR.l2_loss()MKR.loss_kge()MKR.loss_rs()MKR.optimizer_step()MKR.test_epoch_end()MKR.test_step()MKR.test_step_end()MKR.train_kge()MKR.train_rs()MKR.trainingMKR.training_step()MKR.validation_epoch_end()MKR.validation_step()MKR.validation_step_end()
- src.models.collaborativeFiltering package
- Module contents
CollaborativeFilteringCollaborativeFiltering.configure_optimizers()CollaborativeFiltering.eval_epoch_end()CollaborativeFiltering.eval_step()CollaborativeFiltering.eval_step_end()CollaborativeFiltering.forward()CollaborativeFiltering.get_n()CollaborativeFiltering.test_epoch_end()CollaborativeFiltering.test_step()CollaborativeFiltering.test_step_end()CollaborativeFiltering.trainingCollaborativeFiltering.training_step()CollaborativeFiltering.validation_epoch_end()CollaborativeFiltering.validation_step()CollaborativeFiltering.validation_step_end()
- Module contents
- src.models.components package
- src.models.naml package
- Submodules
- src.models.naml.news_encoder module
- src.models.naml.user_encoder module
- Module contents
NAMLNAML.configure_optimizers()NAML.eval_epoch_end()NAML.eval_step()NAML.forward()NAML.get_news_vector()NAML.get_prediction()NAML.get_user_vector()NAML.on_eval_start()NAML.on_test_start()NAML.on_train_start()NAML.on_validation_start()NAML.test_epoch_end()NAML.test_step()NAML.trainingNAML.training_step()NAML.validation_epoch_end()NAML.validation_step()
- src.models.nrms package
- Submodules
- src.models.nrms.news_encoder module
- src.models.nrms.user_encoder module
- Module contents
NRMSNRMS.configure_optimizers()NRMS.eval_epoch_end()NRMS.eval_step()NRMS.forward()NRMS.get_news_vector()NRMS.get_prediction()NRMS.get_user_vector()NRMS.on_eval_start()NRMS.on_test_start()NRMS.on_train_start()NRMS.on_validation_start()NRMS.test_epoch_end()NRMS.test_step()NRMS.trainingNRMS.training_step()NRMS.validation_epoch_end()NRMS.validation_step()
- src.models.rippleNet package
- Module contents
RippleNetRippleNet.configure_optimizers()RippleNet.create_knowledge_graph()RippleNet.create_ripple_set()RippleNet.eval_epoch_end()RippleNet.eval_step()RippleNet.eval_step_end()RippleNet.forward()RippleNet.get_feed_dict()RippleNet.predict()RippleNet.prepare_rating_data_train()RippleNet.test_epoch_end()RippleNet.test_step()RippleNet.test_step_end()RippleNet.trainingRippleNet.training_step()RippleNet.validation_epoch_end()RippleNet.validation_step()RippleNet.validation_step_end()
- Module contents