Index

A

AdditiveAttention (class in src.models.components.attention.additive) (class in src.models.components.utils)

Attention (class in src.models.DKN.attention)

B

BaseDataset (class in src.datamodules.mind.dataset) BehaviorsBERTDataset (class in src.datamodules.mind.dataset)

BehaviorsDataset (class in src.datamodules.mind.dataset) Bias (class in src.models.MKR.layers)

C

CollaborativeFiltering (class in src.models.collaborativeFiltering), [1] compute() (src.models.components.metrics.LogLoss method) (src.models.components.metrics.MRR method) (src.models.components.metrics.Senti method) (src.models.components.metrics.SentiMRR method) configure_optimizers() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.MKR.MKR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] (src.models.rippleNet.RippleNet method), [1]

create_knowledge_graph() (src.models.rippleNet.RippleNet method), [1] create_knowledge_graph_file() (in module src.datamodules.mind.preprocessing) create_rating_file() (in module src.datamodules.mind.preprocessing) create_rating_file_collaborative() (in module src.datamodules.mind.preprocessing) create_ripple_set() (src.models.rippleNet.RippleNet method), [1] CrossCompressUnit (class in src.models.MKR.layers)

D

Dense (class in src.models.MKR.layers) DKN (class in src.models.DKN), [1] DNNClickPredictor (class in src.models.components.click_predictor.DNN) DotProductClickPredictor (class in src.models.components.click_predictor.dot_product)

download_and_extract_glove() (in module src.datamodules.mind.download) download_and_extract_mind() (in module src.datamodules.mind.download) download_and_extract_wikidata_kg() (in module src.datamodules.mind.download) download_path() (in module src.datamodules.components.utils)

E

ElementEncoder (class in src.models.naml.news_encoder) eval_epoch_end() (src.models.collaborativeFiltering.CollaborativeFiltering method) (src.models.DKN.DKN method) (src.models.LSTUR.LSTUR method) (src.models.MKR.MKR method) (src.models.naml.NAML method) (src.models.nrms.NRMS method) (src.models.rippleNet.RippleNet method) eval_step() (src.models.collaborativeFiltering.CollaborativeFiltering method) (src.models.DKN.DKN method) (src.models.LSTUR.LSTUR method) (src.models.MKR.MKR method) (src.models.naml.NAML method) (src.models.nrms.NRMS method) (src.models.rippleNet.RippleNet method)

eval_step_end() (src.models.collaborativeFiltering.CollaborativeFiltering method) (src.models.MKR.MKR method) (src.models.rippleNet.RippleNet method) extract_mind() (in module src.datamodules.mind.download) extras() (in module src.utils)

F

finish() (in module src.utils) forward() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.components.attention.additive.AdditiveAttention method) (src.models.components.attention.multihead_self.MultiHeadSelfAttention method) (src.models.components.attention.multihead_self.ScaledDotProductAttention method) (src.models.components.click_predictor.DNN.DNNClickPredictor method) (src.models.components.click_predictor.dot_product.DotProductClickPredictor method) (src.models.components.simple_dense_net.SimpleDenseNet method) (src.models.components.utils.AdditiveAttention method) (src.models.components.utils.TimeDistributed method) (src.models.DKN.attention.Attention method) (src.models.DKN.DKN method), [1] (src.models.DKN.KCNN.KCNN method) (src.models.LSTUR.LSTUR method), [1] (src.models.LSTUR.news_encoder.NewsEncoder method) (src.models.LSTUR.user_encoder.UserEncoder method) (src.models.MKR.layers.Bias method) (src.models.MKR.layers.CrossCompressUnit method) (src.models.MKR.layers.Dense method) (src.models.MKR.MKR method), [1] (src.models.naml.NAML method), [1] (src.models.naml.news_encoder.ElementEncoder method) (src.models.naml.news_encoder.NewsEncoder method) (src.models.naml.news_encoder.TextEncoder method) (src.models.naml.user_encoder.UserEncoder method) (src.models.nrms.news_encoder.NewsEncoder method) (src.models.nrms.NRMS method), [1] (src.models.nrms.user_encoder.UserEncoder method) (src.models.rippleNet.RippleNet method), [1]

full_state_update (src.models.components.metrics.LogLoss attribute) (src.models.components.metrics.MRR attribute) (src.models.components.metrics.Senti attribute) (src.models.components.metrics.SentiMRR attribute)

G

generate_embeddings() (in module src.datamodules.mind.download) generate_word_embedding() (in module src.datamodules.mind.parse) get_feed_dict() (src.models.rippleNet.RippleNet method), [1] get_logger() (in module src.utils) get_n() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] get_news_vector() (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method) (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] get_prediction() (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1]

get_train_input() (in module src.datamodules.mind.download) get_user_history() (in module src.datamodules.mind.download) get_user_vector() (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] get_valid_input() (in module src.datamodules.mind.download)

K

KCNN (class in src.models.DKN.KCNN)

KGDataset (class in src.datamodules.mind.dataset) (class in src.models.MKR.kGDataset)

L

l2_loss() (src.models.MKR.MKR method) load_glove_matrix() (in module src.datamodules.mind.download) log_hyperparameters() (in module src.utils)

LogLoss (class in src.models.components.metrics) loss_kge() (src.models.MKR.MKR method) loss_rs() (src.models.MKR.MKR method) LSTUR (class in src.models.LSTUR), [1]

M

maybe_download() (in module src.datamodules.components.utils) MINDDataModule (class in src.datamodules.mind.datamodule_Base) MINDDataModuleBERT (class in src.datamodules.mind.datamodule_BERT), [1] MINDDataModuleCollaborativeFiltering (class in src.datamodules.mind.datamodule_CollaborativeFiltering), [1] MINDDataModuleMKR (class in src.datamodules.mind.datamodule_MKR), [1] MINDDataModuleNAML (class in src.datamodules.mind.datamodule_NAML), [1] MINDDataModuleRippleNet (class in src.datamodules.mind.datamodule_RippleNet), [1] MKR (class in src.models.MKR), [1] module src src.baselines src.datamodules src.datamodules.components src.datamodules.components.utils src.datamodules.mind src.datamodules.mind.datamodule_Base src.datamodules.mind.datamodule_BERT src.datamodules.mind.datamodule_CollaborativeFiltering, [1] src.datamodules.mind.datamodule_MKR, [1] src.datamodules.mind.datamodule_NAML, [1] src.datamodules.mind.datamodule_RippleNet, [1] src.datamodules.mind.dataset src.datamodules.mind.download src.datamodules.mind.parse src.datamodules.mind.preprocessing src.models src.models.collaborativeFiltering, [1] src.models.components src.models.components.attention src.models.components.attention.additive src.models.components.attention.multihead_self src.models.components.click_predictor src.models.components.click_predictor.DNN src.models.components.click_predictor.dot_product src.models.components.metrics src.models.components.simple_dense_net src.models.components.utils src.models.DKN, [1] src.models.DKN.attention src.models.DKN.KCNN src.models.LSTUR, [1] src.models.LSTUR.news_encoder src.models.LSTUR.user_encoder src.models.MKR, [1] src.models.MKR.kGDataset src.models.MKR.layers src.models.naml, [1] src.models.naml.news_encoder src.models.naml.user_encoder src.models.nrms, [1] src.models.nrms.news_encoder src.models.nrms.user_encoder src.models.rippleNet, [1] src.testing_pipeline src.training_pipeline src.utils src.vendor

MRR (class in src.models.components.metrics) MultiHeadSelfAttention (class in src.models.components.attention.multihead_self)

N

NAML (class in src.models.naml), [1] news_dataframe() (src.datamodules.mind.datamodule_BERT.MINDDataModuleBERT method) news_dataloader() (src.datamodules.mind.datamodule_NAML.MINDDataModuleNAML method) NewsBERTDataset (class in src.datamodules.mind.dataset)

NewsDataset (class in src.datamodules.mind.dataset) NewsEncoder (class in src.models.LSTUR.news_encoder) (class in src.models.naml.news_encoder) (class in src.models.nrms.news_encoder) NRMS (class in src.models.nrms), [1]

O

on_eval_start() (src.models.DKN.DKN method) (src.models.LSTUR.LSTUR method) (src.models.naml.NAML method) (src.models.nrms.NRMS method) on_test_start() (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1]

on_train_start() (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] on_validation_start() (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] optimizer_step() (src.models.MKR.MKR method), [1]

P

parse_behaviors() (in module src.datamodules.mind.parse) parse_behaviors_bert() (in module src.datamodules.mind.parse) parse_mind() (in module src.datamodules.mind.parse) parse_mind_bert() (in module src.datamodules.mind.parse) parse_news() (in module src.datamodules.mind.parse) parse_news_bert() (in module src.datamodules.mind.parse) predict() (src.models.rippleNet.RippleNet method) prepare() (src.datamodules.mind.datamodule_Base.MINDDataModule method) (src.datamodules.mind.datamodule_CollaborativeFiltering.MINDDataModuleCollaborativeFiltering method), [1] (src.datamodules.mind.datamodule_MKR.MINDDataModuleMKR method), [1] (src.datamodules.mind.datamodule_NAML.MINDDataModuleNAML method) (src.datamodules.mind.datamodule_RippleNet.MINDDataModuleRippleNet method), [1]

prepare_data() (src.datamodules.mind.datamodule_BERT.MINDDataModuleBERT method), [1] (src.datamodules.mind.datamodule_CollaborativeFiltering.MINDDataModuleCollaborativeFiltering method), [1] (src.datamodules.mind.datamodule_MKR.MINDDataModuleMKR method), [1] (src.datamodules.mind.datamodule_RippleNet.MINDDataModuleRippleNet method), [1] prepare_numpy_data() (in module src.datamodules.mind.preprocessing) prepare_numpy_kg() (in module src.datamodules.mind.preprocessing) prepare_rating_data_train() (src.models.rippleNet.RippleNet method), [1] print_config() (in module src.utils)

R

RatingsDataset (class in src.datamodules.mind.dataset) read_clickhistory() (in module src.datamodules.mind.download)

read_news() (in module src.datamodules.mind.download) read_news_ids() (in module src.datamodules.mind.download) RippleNet (class in src.models.rippleNet), [1]

S

ScaledDotProductAttention (class in src.models.components.attention.multihead_self) Senti (class in src.models.components.metrics) SentiMRR (class in src.models.components.metrics) setup() (src.datamodules.mind.datamodule_Base.MINDDataModule method) (src.datamodules.mind.datamodule_BERT.MINDDataModuleBERT method), [1] (src.datamodules.mind.datamodule_CollaborativeFiltering.MINDDataModuleCollaborativeFiltering method), [1] (src.datamodules.mind.datamodule_MKR.MINDDataModuleMKR method), [1] (src.datamodules.mind.datamodule_NAML.MINDDataModuleNAML method), [1] (src.datamodules.mind.datamodule_RippleNet.MINDDataModuleRippleNet method), [1] SimpleDenseNet (class in src.models.components.simple_dense_net) src module src.baselines module src.datamodules module src.datamodules.components module src.datamodules.components.utils module src.datamodules.mind module src.datamodules.mind.datamodule_Base module src.datamodules.mind.datamodule_BERT module src.datamodules.mind.datamodule_CollaborativeFiltering module, [1] src.datamodules.mind.datamodule_MKR module, [1] src.datamodules.mind.datamodule_NAML module, [1] src.datamodules.mind.datamodule_RippleNet module, [1] src.datamodules.mind.dataset module src.datamodules.mind.download module src.datamodules.mind.parse module src.datamodules.mind.preprocessing module src.models module src.models.collaborativeFiltering module, [1] src.models.components module src.models.components.attention module src.models.components.attention.additive module src.models.components.attention.multihead_self module

src.models.components.click_predictor module src.models.components.click_predictor.DNN module src.models.components.click_predictor.dot_product module src.models.components.metrics module src.models.components.simple_dense_net module src.models.components.utils module src.models.DKN module, [1] src.models.DKN.attention module src.models.DKN.KCNN module src.models.LSTUR module, [1] src.models.LSTUR.news_encoder module src.models.LSTUR.user_encoder module src.models.MKR module, [1] src.models.MKR.kGDataset module src.models.MKR.layers module src.models.naml module, [1] src.models.naml.news_encoder module src.models.naml.user_encoder module src.models.nrms module, [1] src.models.nrms.news_encoder module src.models.nrms.user_encoder module src.models.rippleNet module, [1] src.testing_pipeline module src.training_pipeline module src.utils module src.vendor module

T

test() (in module src.testing_pipeline) test_dataloader() (src.datamodules.mind.datamodule_Base.MINDDataModule method) (src.datamodules.mind.datamodule_BERT.MINDDataModuleBERT method), [1] (src.datamodules.mind.datamodule_NAML.MINDDataModuleNAML method), [1] test_epoch_end() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.MKR.MKR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] (src.models.rippleNet.RippleNet method), [1] test_step() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.MKR.MKR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] (src.models.rippleNet.RippleNet method), [1] test_step_end() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.MKR.MKR method), [1] (src.models.rippleNet.RippleNet method), [1] TextEncoder (class in src.models.naml.news_encoder) TimeDistributed (class in src.models.components.utils) to() (src.datamodules.mind.dataset.NewsDataset method) train() (in module src.training_pipeline) train_dataloader() (src.datamodules.mind.datamodule_Base.MINDDataModule method) (src.datamodules.mind.datamodule_BERT.MINDDataModuleBERT method), [1] (src.datamodules.mind.datamodule_CollaborativeFiltering.MINDDataModuleCollaborativeFiltering method), [1] (src.datamodules.mind.datamodule_MKR.MINDDataModuleMKR method), [1] (src.datamodules.mind.datamodule_RippleNet.MINDDataModuleRippleNet method), [1] train_kge() (src.models.MKR.MKR method) train_rs() (src.models.MKR.MKR method) training (src.models.collaborativeFiltering.CollaborativeFiltering attribute) (src.models.components.attention.additive.AdditiveAttention attribute) (src.models.components.attention.multihead_self.MultiHeadSelfAttention attribute) (src.models.components.attention.multihead_self.ScaledDotProductAttention attribute) (src.models.components.click_predictor.DNN.DNNClickPredictor attribute) (src.models.components.click_predictor.dot_product.DotProductClickPredictor attribute) (src.models.components.simple_dense_net.SimpleDenseNet attribute) (src.models.components.utils.AdditiveAttention attribute) (src.models.components.utils.TimeDistributed attribute) (src.models.DKN.attention.Attention attribute) (src.models.DKN.DKN attribute) (src.models.DKN.KCNN.KCNN attribute) (src.models.LSTUR.LSTUR attribute) (src.models.LSTUR.news_encoder.NewsEncoder attribute) (src.models.LSTUR.user_encoder.UserEncoder attribute) (src.models.MKR.layers.Bias attribute) (src.models.MKR.layers.CrossCompressUnit attribute) (src.models.MKR.layers.Dense attribute) (src.models.MKR.MKR attribute) (src.models.naml.NAML attribute) (src.models.naml.news_encoder.ElementEncoder attribute) (src.models.naml.news_encoder.NewsEncoder attribute) (src.models.naml.news_encoder.TextEncoder attribute) (src.models.naml.user_encoder.UserEncoder attribute) (src.models.nrms.news_encoder.NewsEncoder attribute) (src.models.nrms.NRMS attribute) (src.models.nrms.user_encoder.UserEncoder attribute) (src.models.rippleNet.RippleNet attribute)

training_step() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.MKR.MKR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] (src.models.rippleNet.RippleNet method), [1] transform_entity_embedding() (in module src.datamodules.mind.parse)

U

unzip_file() (in module src.datamodules.components.utils) update() (src.models.components.metrics.LogLoss method) (src.models.components.metrics.MRR method) (src.models.components.metrics.Senti method) (src.models.components.metrics.SentiMRR method)

user_dataloader() (src.datamodules.mind.datamodule_NAML.MINDDataModuleNAML method) UserDataset (class in src.datamodules.mind.dataset) UserEncoder (class in src.models.LSTUR.user_encoder) (class in src.models.naml.user_encoder) (class in src.models.nrms.user_encoder)

V

val_dataloader() (src.datamodules.mind.datamodule_Base.MINDDataModule method) (src.datamodules.mind.datamodule_BERT.MINDDataModuleBERT method), [1] (src.datamodules.mind.datamodule_NAML.MINDDataModuleNAML method), [1] validation_epoch_end() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.MKR.MKR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] (src.models.rippleNet.RippleNet method), [1]

validation_step() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.DKN.DKN method), [1] (src.models.LSTUR.LSTUR method), [1] (src.models.MKR.MKR method), [1] (src.models.naml.NAML method), [1] (src.models.nrms.NRMS method), [1] (src.models.rippleNet.RippleNet method), [1] validation_step_end() (src.models.collaborativeFiltering.CollaborativeFiltering method), [1] (src.models.MKR.MKR method), [1] (src.models.rippleNet.RippleNet method), [1]

W

word_tokenize() (in module src.datamodules.mind.download)