Index

A

add_reverse_relation() (neuralkg_ind.data.DataPreprocess.RevSampler method) add_reverse_triples() (neuralkg_ind.data.DataPreprocess.RevSampler method) add_to_argparse() (neuralkg_ind.data.base_data_module.BaseDataModule static method) (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel static method) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel static method) (neuralkg_ind.lit_model.KGELitModel.KGELitModel static method) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel static method) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel static method) adj_matrix (neuralkg_ind.data.Sampler.KBATSampler attribute) Adv_Loss (class in neuralkg_ind.loss.Adv_Loss) AdvSampler (class in neuralkg_ind.data.Sampler) aggragate_emb() (neuralkg_ind.model.GNNModel.SEGNN.SEGNN method) AggregateConv() (neuralkg_ind.model.GNNModel.RMPI.RMPI method) Aggregator (class in neuralkg_ind.model.GNNModel.layer) aggregator (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RGCN attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) all_true_triples (neuralkg_ind.data.DataPreprocess.KGData attribute) allow_zero_length_dataloader_with_multiple_devices (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel attribute) (neuralkg_ind.lit_model.IterELitModel.IterELitModel attribute) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg_ind.lit_model.KGELitModel.KGELitModel attribute) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel attribute) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg_ind.lit_model.RugELitModel.RugELitModel attribute) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel attribute) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel attribute) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel attribute) AllSampler (class in neuralkg_ind.data.Sampler) apply_node_func() (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv method) args (neuralkg_ind.data.DataPreprocess.GRData attribute) (neuralkg_ind.data.DataPreprocess.KGData attribute) (neuralkg_ind.data.Sampler.TestRMPISampler_auc attribute) (neuralkg_ind.data.Sampler.TestRMPISampler_hit attribute) (neuralkg_ind.data.Sampler.TestSampler_auc attribute) (neuralkg_ind.data.Sampler.TestSampler_hit attribute) (neuralkg_ind.data.Sampler.ValidRMPISampler attribute) (neuralkg_ind.data.Sampler.ValidSampler attribute) (neuralkg_ind.loss.Adv_Loss.Adv_Loss attribute) (neuralkg_ind.model.GNNModel.CompGCN.CompGCN attribute) (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE attribute) (neuralkg_ind.model.GNNModel.Grail.Grail attribute) (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.Grail.RGCN attribute) (neuralkg_ind.model.GNNModel.KBAT.KBAT attribute) (neuralkg_ind.model.GNNModel.MorsE.EntInit attribute) (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) (neuralkg_ind.model.GNNModel.MorsE.MorsE attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RGCN attribute) (neuralkg_ind.model.GNNModel.RGCN.RGCN attribute) (neuralkg_ind.model.GNNModel.RMPI.RMPI attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.SNRI attribute) (neuralkg_ind.model.GNNModel.XTransE.XTransE attribute) (neuralkg_ind.model.KGEModel.BoxE.BoxE attribute) (neuralkg_ind.model.KGEModel.ConvE.ConvE attribute) (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute) (neuralkg_ind.model.KGEModel.DistMult.DistMult attribute) (neuralkg_ind.model.KGEModel.DualE.DualE attribute) (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) (neuralkg_ind.model.KGEModel.PairRE.PairRE attribute) (neuralkg_ind.model.KGEModel.RotatE.RotatE attribute) (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) (neuralkg_ind.model.KGEModel.TransE.TransE attribute) (neuralkg_ind.model.KGEModel.TransH.TransH attribute) (neuralkg_ind.model.KGEModel.TransR.TransR attribute) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg_ind.model.RuleModel.IterE.IterE attribute) (neuralkg_ind.model.RuleModel.RugE.RugE attribute)

attn_rel_emb (neuralkg_ind.model.GNNModel.Grail.RGCN attribute) attn_rel_emb_dim (neuralkg_ind.model.GNNModel.Grail.RGCN attribute)

B

backward() (neuralkg_ind.model.GNNModel.CoMPILE.MySpMM static method) (neuralkg_ind.model.GNNModel.KBAT.SpecialSpmmFunctionFinal static method) BaseDataModule (class in neuralkg_ind.data.base_data_module) BaseGraph (class in neuralkg_ind.data.DataPreprocess) BaseLitModel (class in neuralkg_ind.lit_model.BaseLitModel) BaseMeta (class in neuralkg_ind.data.DataPreprocess) BaseSampler (class in neuralkg_ind.data.DataPreprocess) basiclayer (neuralkg_ind.model.GNNModel.Grail.RGCN attribute) (neuralkg_ind.model.GNNModel.MorsE.RGCN attribute) batch_size (neuralkg_ind.data.RuleDataLoader.RuleDataLoader attribute) batch_subgraph() (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE method) BatchGRU (class in neuralkg_ind.model.GNNModel.layer) BernSampler (class in neuralkg_ind.data.Sampler)

bfs() (neuralkg_ind.data.Sampler.KBATSampler method) bfs_relational() (in module neuralkg_ind.utils.tools) bias (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) BoxE (class in neuralkg_ind.model.KGEModel.BoxE) build_graph() (neuralkg_ind.data.Sampler.GraphSampler method) build_hidden_layer() (neuralkg_ind.model.GNNModel.CompGCN.CompGCN method) (neuralkg_ind.model.GNNModel.Grail.RGCN method) (neuralkg_ind.model.GNNModel.model.Model method) (neuralkg_ind.model.GNNModel.MorsE.RGCN method) (neuralkg_ind.model.GNNModel.RGCN.RGCN method) build_model() (neuralkg_ind.model.GNNModel.model.Model method)

C

c (neuralkg_ind.model.GNNModel.SNRI.Discriminator attribute) cal_Con_score() (neuralkg_ind.model.GNNModel.KBAT.KBAT method) cal_GAT_score() (neuralkg_ind.model.GNNModel.KBAT.KBAT method) calc_bern() (neuralkg_ind.data.Sampler.BernSampler method) calc_freq() (neuralkg_ind.data.Sampler.AdvSampler method) calc_ranks() (in module neuralkg_ind.eval_task.link_prediction) calc_ranks_SEGNN() (in module neuralkg_ind.eval_task.link_prediction) calc_score() (neuralkg_ind.model.KGEModel.BoxE.BoxE method) classification() (in module neuralkg_ind.eval_task.triple_classification) collate_fn() (neuralkg_ind.data.DataPreprocess.KGEEvalData static method) comp() (neuralkg_ind.model.GNNModel.CompGCN.CompGCNCov method) comp_deg_norm() (neuralkg_ind.data.Sampler.GraphSampler method) comp_hrt_emb() (neuralkg_ind.model.GNNModel.SNRI.SNRI method) comp_ht_emb() (neuralkg_ind.model.GNNModel.SNRI.SNRI method) CompGCN (class in neuralkg_ind.model.GNNModel.CompGCN) CompGCNCov (class in neuralkg_ind.model.GNNModel.CompGCN) CompGCNLitModel (class in neuralkg_ind.lit_model.CompGCNLitModel) CompGCNSampler (class in neuralkg_ind.data.Sampler) CompGCNTestSampler (class in neuralkg_ind.data.Sampler) CoMPILE (class in neuralkg_ind.model.GNNModel.CoMPILE) CoMPILEConv() (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE method) CompLayer (class in neuralkg_ind.model.GNNModel.SEGNN) ComplEx (class in neuralkg_ind.model.KGEModel.ComplEx) ComplEx_NNE_AER (class in neuralkg_ind.model.RuleModel.ComplEx_NNE_AER) ComplEx_NNE_AER_Loss (class in neuralkg_ind.loss.ComplEx_NNE_AER_Loss) conc (neuralkg_ind.model.GNNModel.RMPI.RMPI attribute) concat() (neuralkg_ind.model.GNNModel.CompGCN.CompGCN method) (neuralkg_ind.model.GNNModel.SEGNN.SEGNN method) (neuralkg_ind.model.KGEModel.ConvE.ConvE method) Config (class in neuralkg_ind.data.base_data_module) (class in neuralkg_ind.lit_model.BaseLitModel)

configure_optimizers() (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel method) (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel method) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel method) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel method) (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel method) (neuralkg_ind.lit_model.KGELitModel.KGELitModel method) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel method) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg_ind.lit_model.RugELitModel.RugELitModel method) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel method) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel method) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel method) construct_kg() (neuralkg_ind.data.Sampler.SEGNNTrainProcess method) ConvE (class in neuralkg_ind.model.KGEModel.ConvE) ConvELitModel (class in neuralkg_ind.lit_model.ConvELitModel) ConvSampler (class in neuralkg_ind.data.Sampler) corrupt_head() (neuralkg_ind.data.DataPreprocess.BaseSampler method) (neuralkg_ind.data.DataPreprocess.RevSampler method) corrupt_tail() (neuralkg_ind.data.DataPreprocess.BaseSampler method) (neuralkg_ind.data.DataPreprocess.RevSampler method) count_frequency() (neuralkg_ind.data.DataPreprocess.KGData static method) Cross_Entropy_Loss (class in neuralkg_ind.loss.Cross_Entropy_Loss) cross_sampling_flag (neuralkg_ind.data.Sampler.UniSampler attribute) CrossE (class in neuralkg_ind.model.KGEModel.CrossE) CrossE_Loss (class in neuralkg_ind.loss.CrossE_Loss) CrossELitModel (class in neuralkg_ind.lit_model.CrossELitModel) CrossESampler (class in neuralkg_ind.data.Sampler)

D

data2pkl() (in module neuralkg_ind.utils.tools) dataset (neuralkg_ind.data.RuleDataLoader.RuleDataLoader attribute) db_name_neg (neuralkg_ind.data.DataPreprocess.GRData attribute) db_name_pos (neuralkg_ind.data.DataPreprocess.GRData attribute) deserialize() (in module neuralkg_ind.utils.tools)

deserialize_RMPI() (in module neuralkg_ind.utils.tools) Discriminator (class in neuralkg_ind.model.GNNModel.SNRI) dist() (neuralkg_ind.model.KGEModel.BoxE.BoxE method) DistMult (class in neuralkg_ind.model.KGEModel.DistMult) drop_last (neuralkg_ind.data.RuleDataLoader.RuleDataLoader attribute) DualE (class in neuralkg_ind.model.KGEModel.DualE)

E

edge_dropout (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) EdgeLayer (class in neuralkg_ind.model.GNNModel.SEGNN) emb_dim (neuralkg_ind.model.GNNModel.Grail.RGCN attribute) (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) (neuralkg_ind.model.GNNModel.MorsE.RGCN attribute) embedding_range (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) (neuralkg_ind.model.KGEModel.DistMult.DistMult attribute) (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) (neuralkg_ind.model.KGEModel.PairRE.PairRE attribute) (neuralkg_ind.model.KGEModel.RotatE.RotatE attribute) (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) (neuralkg_ind.model.KGEModel.TransE.TransE attribute) (neuralkg_ind.model.KGEModel.TransH.TransH attribute) (neuralkg_ind.model.KGEModel.TransR.TransR attribute) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg_ind.model.RuleModel.IterE.IterE attribute) (neuralkg_ind.model.RuleModel.RugE.RugE attribute) ent2id (neuralkg_ind.data.DataPreprocess.KGData attribute) ent_emb (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute) (neuralkg_ind.model.KGEModel.DistMult.DistMult attribute) (neuralkg_ind.model.KGEModel.DualE.DualE attribute) (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) (neuralkg_ind.model.KGEModel.PairRE.PairRE attribute) (neuralkg_ind.model.KGEModel.RotatE.RotatE attribute) (neuralkg_ind.model.KGEModel.TransE.TransE attribute) (neuralkg_ind.model.KGEModel.TransH.TransH attribute) (neuralkg_ind.model.KGEModel.TransR.TransR attribute) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg_ind.model.RuleModel.IterE.IterE attribute) (neuralkg_ind.model.RuleModel.RugE.RugE attribute)

ent_h_emb (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) ent_init (neuralkg_ind.model.GNNModel.MorsE.MorsE attribute) ent_padding (neuralkg_ind.model.GNNModel.SNRI.SNRI attribute) ent_t_emb (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) EntInit (class in neuralkg_ind.model.GNNModel.MorsE) entity (neuralkg_ind.data.Sampler.GraphSampler attribute) epsilon (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) (neuralkg_ind.model.KGEModel.DistMult.DistMult attribute) (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) (neuralkg_ind.model.KGEModel.PairRE.PairRE attribute) (neuralkg_ind.model.KGEModel.RotatE.RotatE attribute) (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) (neuralkg_ind.model.KGEModel.TransE.TransE attribute) (neuralkg_ind.model.KGEModel.TransH.TransH attribute) (neuralkg_ind.model.KGEModel.TransR.TransR attribute) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg_ind.model.RuleModel.IterE.IterE attribute) (neuralkg_ind.model.RuleModel.RugE.RugE attribute) extract_save_subgraph() (in module neuralkg_ind.utils.tools)

F

forward() (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel method) (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel method) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel method) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel method) (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel method) (neuralkg_ind.lit_model.KGELitModel.KGELitModel method) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel method) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg_ind.lit_model.RugELitModel.RugELitModel method) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel method) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel method) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel method) (neuralkg_ind.loss.Adv_Loss.Adv_Loss method) (neuralkg_ind.loss.ComplEx_NNE_AER_Loss.ComplEx_NNE_AER_Loss method) (neuralkg_ind.loss.Cross_Entropy_Loss.Cross_Entropy_Loss method) (neuralkg_ind.loss.CrossE_Loss.CrossE_Loss method) (neuralkg_ind.loss.KBAT_Loss.KBAT_Loss method) (neuralkg_ind.loss.Logsig_Loss.Logsig_Loss method) (neuralkg_ind.loss.Margin_Loss.Margin_Loss method) (neuralkg_ind.loss.RGCN_Loss.RGCN_Loss method) (neuralkg_ind.loss.RugE_Loss.RugE_Loss method) (neuralkg_ind.loss.SimplE_Loss.SimplE_Loss method) (neuralkg_ind.loss.Softplus_Loss.Softplus_Loss method) (neuralkg_ind.model.GNNModel.CompGCN.CompGCN method) (neuralkg_ind.model.GNNModel.CompGCN.CompGCNCov method) (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE method) (neuralkg_ind.model.GNNModel.CoMPILE.MySpMM static method) (neuralkg_ind.model.GNNModel.Grail.Grail method) (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv method) (neuralkg_ind.model.GNNModel.Grail.RGCN method) (neuralkg_ind.model.GNNModel.KBAT.GraphAttentionLayer method) (neuralkg_ind.model.GNNModel.KBAT.KBAT method) (neuralkg_ind.model.GNNModel.KBAT.SpecialSpmmFinal method) (neuralkg_ind.model.GNNModel.KBAT.SpecialSpmmFunctionFinal static method) (neuralkg_ind.model.GNNModel.layer.Aggregator method) (neuralkg_ind.model.GNNModel.layer.BatchGRU method) (neuralkg_ind.model.GNNModel.layer.GNNLayer method) (neuralkg_ind.model.GNNModel.model.Model method) (neuralkg_ind.model.GNNModel.MorsE.EntInit method) (neuralkg_ind.model.GNNModel.MorsE.KGEModel method) (neuralkg_ind.model.GNNModel.MorsE.MorsE method) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv method) (neuralkg_ind.model.GNNModel.MorsE.RGCN method) (neuralkg_ind.model.GNNModel.RGCN.RelGraphConv method) (neuralkg_ind.model.GNNModel.RGCN.RGCN method) (neuralkg_ind.model.GNNModel.RMPI.RMPI method) (neuralkg_ind.model.GNNModel.SEGNN.CompLayer method) (neuralkg_ind.model.GNNModel.SEGNN.EdgeLayer method) (neuralkg_ind.model.GNNModel.SEGNN.NodeLayer method) (neuralkg_ind.model.GNNModel.SEGNN.SEGNN method) (neuralkg_ind.model.GNNModel.SNRI.Discriminator method) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv method) (neuralkg_ind.model.GNNModel.SNRI.SNRI method) (neuralkg_ind.model.GNNModel.XTransE.XTransE method) (neuralkg_ind.model.KGEModel.BoxE.BoxE method) (neuralkg_ind.model.KGEModel.ComplEx.ComplEx method) (neuralkg_ind.model.KGEModel.ConvE.ConvE method) (neuralkg_ind.model.KGEModel.CrossE.CrossE method) (neuralkg_ind.model.KGEModel.DistMult.DistMult method) (neuralkg_ind.model.KGEModel.DualE.DualE method) (neuralkg_ind.model.KGEModel.HAKE.HAKE method) (neuralkg_ind.model.KGEModel.model.Model method) (neuralkg_ind.model.KGEModel.PairRE.PairRE method) (neuralkg_ind.model.KGEModel.RotatE.RotatE method) (neuralkg_ind.model.KGEModel.SimplE.SimplE method) (neuralkg_ind.model.KGEModel.TransE.TransE method) (neuralkg_ind.model.KGEModel.TransH.TransH method) (neuralkg_ind.model.KGEModel.TransR.TransR method) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg_ind.model.RuleModel.IterE.IterE method) (neuralkg_ind.model.RuleModel.model.Model method) (neuralkg_ind.model.RuleModel.RugE.RugE method)

forward_Con() (neuralkg_ind.model.GNNModel.KBAT.KBAT method) forward_GAT() (neuralkg_ind.model.GNNModel.KBAT.KBAT method)

G

gen_meta_subgraph_datasets() (in module neuralkg_ind.utils.tools) gen_subgraph_datasets() (in module neuralkg_ind.utils.tools) generate_ind_test() (neuralkg_ind.data.DataPreprocess.BaseGraph method) generate_new_train_triples() (neuralkg_ind.model.RuleModel.IterE.IterE method) generate_train() (neuralkg_ind.data.DataPreprocess.GRData method) generate_valid() (neuralkg_ind.data.DataPreprocess.GRData method) get_all_true_triples() (neuralkg_ind.data.DataPreprocess.BaseSampler method) (neuralkg_ind.data.DataPreprocess.RevSampler method) get_auc() (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel method) get_average_meta_subgraph_size() (in module neuralkg_ind.utils.tools) get_average_subgraph_size() (in module neuralkg_ind.utils.tools) get_axiom() (neuralkg_ind.model.RuleModel.IterE.IterE method) get_batch_nhop_neighbors_all() (neuralkg_ind.data.Sampler.KBATSampler method) get_config() (neuralkg_ind.data.base_data_module.BaseDataModule method) get_data_config() (neuralkg_ind.data.KGDataModule.KGDataModule method) get_edge_count() (in module neuralkg_ind.utils.tools) get_emb() (neuralkg_ind.model.KGEModel.SimplE.SimplE method) get_ent_emb() (neuralkg_ind.model.GNNModel.MorsE.MorsE method) get_g() (in module neuralkg_ind.utils.tools) get_g_bidir() (in module neuralkg_ind.utils.tools) get_h2rt_t2hr_from_train() (neuralkg_ind.data.DataPreprocess.KGData method) (neuralkg_ind.data.Sampler.SEGNNTrainProcess method) get_hr2t_rt2h() (in module neuralkg_ind.utils.tools) get_hr2t_rt2h_from_all() (neuralkg_ind.data.Sampler.CompGCNTestSampler method) (neuralkg_ind.data.Sampler.GraphTestSampler method) (neuralkg_ind.data.Sampler.SEGNNTestSampler method) (neuralkg_ind.data.Sampler.TestSampler method) get_hr2t_rt2h_from_train() (neuralkg_ind.data.DataPreprocess.KGData method) get_hr2t_rt2h_sup_que() (in module neuralkg_ind.utils.tools) get_hr_trian() (neuralkg_ind.data.DataPreprocess.KGData method) get_id() (neuralkg_ind.data.DataPreprocess.KGData method) get_indtest_test_dataset_and_train_g() (in module neuralkg_ind.utils.tools) get_intest_train_g() (neuralkg_ind.model.GNNModel.MorsE.MorsE method) get_kg() (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel method) get_label() (neuralkg_ind.data.DataPreprocess.KGEEvalData method) get_logits() (neuralkg_ind.model.GNNModel.SNRI.SNRI method) get_neg_samples_replacing_head_tail() (neuralkg_ind.data.DataPreprocess.BaseGraph method) get_neighbor_nodes() (in module neuralkg_ind.utils.tools) get_neighbors() (in module neuralkg_ind.utils.tools) (neuralkg_ind.data.Sampler.KBATSampler method) get_num_rel() (neuralkg_ind.model.GNNModel.MorsE.MorsE method) get_onehot_label() (neuralkg_ind.data.Sampler.SEGNNTrainProcess method) get_param() (in module neuralkg_ind.utils.tools) (neuralkg_ind.model.GNNModel.CompGCN.CompGCNCov method) get_results() (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel method) get_rule() (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg_ind.model.RuleModel.IterE.IterE method) get_sampling() (neuralkg_ind.data.Sampler.SEGNNTrainProcess method) get_sampling_keys() (neuralkg_ind.data.Sampler.CompGCNSampler method) (neuralkg_ind.data.Sampler.CompGCNTestSampler method) (neuralkg_ind.data.Sampler.GraphSampler method) (neuralkg_ind.data.Sampler.GraphTestSampler method) (neuralkg_ind.data.Sampler.KBATSampler method) (neuralkg_ind.data.Sampler.MetaSampler method) (neuralkg_ind.data.Sampler.RMPISampler method) (neuralkg_ind.data.Sampler.SEGNNTestSampler method) (neuralkg_ind.data.Sampler.SubSampler method) (neuralkg_ind.data.Sampler.TestMetaSampler_auc method) (neuralkg_ind.data.Sampler.TestMetaSampler_hit method) (neuralkg_ind.data.Sampler.TestRMPISampler_auc method) (neuralkg_ind.data.Sampler.TestRMPISampler_hit method) (neuralkg_ind.data.Sampler.TestSampler method) (neuralkg_ind.data.Sampler.TestSampler_hit method) (neuralkg_ind.data.Sampler.UniSampler method) (neuralkg_ind.data.Sampler.ValidMetaSampler method) (neuralkg_ind.data.Sampler.ValidRMPISampler method) (neuralkg_ind.data.Sampler.ValidSampler method) (neuralkg_ind.data.Sampler.XTransESampler method)

get_score() (neuralkg_ind.model.GNNModel.CompGCN.CompGCN method) (neuralkg_ind.model.GNNModel.KBAT.KBAT method) (neuralkg_ind.model.GNNModel.MorsE.MorsE method) (neuralkg_ind.model.GNNModel.RGCN.RGCN method) (neuralkg_ind.model.GNNModel.XTransE.XTransE method) (neuralkg_ind.model.KGEModel.BoxE.BoxE method) (neuralkg_ind.model.KGEModel.ComplEx.ComplEx method) (neuralkg_ind.model.KGEModel.ConvE.ConvE method) (neuralkg_ind.model.KGEModel.CrossE.CrossE method) (neuralkg_ind.model.KGEModel.DistMult.DistMult method) (neuralkg_ind.model.KGEModel.DualE.DualE method) (neuralkg_ind.model.KGEModel.HAKE.HAKE method) (neuralkg_ind.model.KGEModel.PairRE.PairRE method) (neuralkg_ind.model.KGEModel.RotatE.RotatE method) (neuralkg_ind.model.KGEModel.SimplE.SimplE method) (neuralkg_ind.model.KGEModel.TransE.TransE method) (neuralkg_ind.model.KGEModel.TransH.TransH method) (neuralkg_ind.model.KGEModel.TransR.TransR method) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg_ind.model.RuleModel.IterE.IterE method) (neuralkg_ind.model.RuleModel.RugE.RugE method) get_subgraphs() (neuralkg_ind.data.Sampler.TestRMPISampler_hit method) (neuralkg_ind.data.Sampler.TestSampler_hit method) get_test() (neuralkg_ind.data.DataPreprocess.BaseGraph method) (neuralkg_ind.data.DataPreprocess.BaseMeta method) (neuralkg_ind.data.DataPreprocess.BaseSampler method) (neuralkg_ind.data.DataPreprocess.RevSampler method) (neuralkg_ind.data.Sampler.SEGNNTrainSampler method) get_train() (neuralkg_ind.data.DataPreprocess.BaseGraph method) (neuralkg_ind.data.DataPreprocess.BaseMeta method) (neuralkg_ind.data.DataPreprocess.BaseSampler method) (neuralkg_ind.data.DataPreprocess.RevSampler method) (neuralkg_ind.data.Sampler.SEGNNTrainSampler method) get_train_bs() (neuralkg_ind.data.KGDataModule.KGDataModule method) get_triples_id() (neuralkg_ind.data.DataPreprocess.KGData method) get_unique_entity() (neuralkg_ind.data.Sampler.KBATSampler method) get_valid() (neuralkg_ind.data.DataPreprocess.BaseGraph method) (neuralkg_ind.data.DataPreprocess.BaseMeta method) (neuralkg_ind.data.DataPreprocess.BaseSampler method) (neuralkg_ind.data.DataPreprocess.RevSampler method) (neuralkg_ind.data.Sampler.SEGNNTrainSampler method) gnn (neuralkg_ind.model.GNNModel.Grail.Grail attribute) (neuralkg_ind.model.GNNModel.SNRI.SNRI attribute) gnn_spmm() (in module neuralkg_ind.model.GNNModel.CoMPILE) GNNLayer (class in neuralkg_ind.model.GNNModel.layer) Grail (class in neuralkg_ind.model.GNNModel.Grail) graph (neuralkg_ind.data.DataPreprocess.GRData attribute) (neuralkg_ind.data.Sampler.CompGCNSampler attribute) (neuralkg_ind.data.Sampler.GraphSampler attribute) (neuralkg_ind.data.Sampler.KBATSampler attribute) GraphAttentionLayer (class in neuralkg_ind.model.GNNModel.KBAT) GraphSampler (class in neuralkg_ind.data.Sampler) GraphTestSampler (class in neuralkg_ind.data.Sampler) GRData (class in neuralkg_ind.data.DataPreprocess) GroundAllRules (class in neuralkg_ind.data.Grounding) groundRule() (neuralkg_ind.data.Grounding.GroundAllRules method) GRUAggregator (class in neuralkg_ind.model.GNNModel.layer)

H

h2rt_train (neuralkg_ind.data.DataPreprocess.KGData attribute) h_mask (neuralkg_ind.data.Sampler.XTransESampler attribute) h_mi (neuralkg_ind.model.GNNModel.SNRI.Discriminator attribute) h_neighbor (neuralkg_ind.data.Sampler.XTransESampler attribute) h_pl (neuralkg_ind.model.GNNModel.SNRI.Discriminator attribute) h_weighted_vector (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute) HAKE (class in neuralkg_ind.model.KGEModel.HAKE) has_attn (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.Grail.RGCN attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute)

head_batch() (neuralkg_ind.data.DataPreprocess.BaseSampler method) (neuralkg_ind.data.DataPreprocess.RevSampler method) head_predict() (in module neuralkg_ind.eval_task.link_prediction) head_predict_SEGNN() (in module neuralkg_ind.eval_task.link_prediction) hidden_size (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE attribute) hr2t (neuralkg_ind.data.DataPreprocess.KGEEvalData attribute) hr2t_all (neuralkg_ind.data.Sampler.CompGCNTestSampler attribute) (neuralkg_ind.data.Sampler.GraphTestSampler attribute) (neuralkg_ind.data.Sampler.TestSampler attribute) hr2t_train (neuralkg_ind.data.DataPreprocess.KGData attribute) (neuralkg_ind.data.DataPreprocess.RevSampler attribute) hr_bias (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute)

I

id2ent (neuralkg_ind.data.DataPreprocess.KGData attribute) id2entity (neuralkg_ind.data.DataPreprocess.GRData attribute) id2rel (neuralkg_ind.data.DataPreprocess.KGData attribute) id2relation (neuralkg_ind.data.DataPreprocess.GRData attribute) import_class() (in module neuralkg_ind.utils.tools) incidence_matrix() (in module neuralkg_ind.utils.tools) ind_predict() (in module neuralkg_ind.eval_task.link_prediction) indGNNLitModel (class in neuralkg_ind.lit_model.indGNNLitModel) init_ConvKB_emb() (neuralkg_ind.model.GNNModel.KBAT.KBAT method) init_emb() (neuralkg_ind.model.GNNModel.CompGCN.CompGCN method) (neuralkg_ind.model.GNNModel.Grail.RGCN method) (neuralkg_ind.model.GNNModel.model.Model method) (neuralkg_ind.model.GNNModel.RGCN.RGCN method) (neuralkg_ind.model.GNNModel.XTransE.XTransE method) (neuralkg_ind.model.KGEModel.BoxE.BoxE method) (neuralkg_ind.model.KGEModel.ComplEx.ComplEx method) (neuralkg_ind.model.KGEModel.ConvE.ConvE method) (neuralkg_ind.model.KGEModel.CrossE.CrossE method) (neuralkg_ind.model.KGEModel.DistMult.DistMult method) (neuralkg_ind.model.KGEModel.HAKE.HAKE method) (neuralkg_ind.model.KGEModel.model.Model method) (neuralkg_ind.model.KGEModel.PairRE.PairRE method) (neuralkg_ind.model.KGEModel.RotatE.RotatE method) (neuralkg_ind.model.KGEModel.SimplE.SimplE method) (neuralkg_ind.model.KGEModel.TransE.TransE method) (neuralkg_ind.model.KGEModel.TransH.TransH method) (neuralkg_ind.model.KGEModel.TransR.TransR method) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg_ind.model.RuleModel.IterE.IterE method) (neuralkg_ind.model.RuleModel.model.Model method) (neuralkg_ind.model.RuleModel.RugE.RugE method)

init_ent_emb_matrix() (neuralkg_ind.model.GNNModel.SNRI.SNRI method) init_GAT_emb() (neuralkg_ind.model.GNNModel.KBAT.KBAT method) init_label() (neuralkg_ind.data.Sampler.CrossESampler method) init_weights() (neuralkg_ind.model.KGEModel.DualE.DualE method) inp_dim (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.Grail.RGCN attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RGCN attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) intialize_worker() (in module neuralkg_ind.utils.tools) is_input_layer (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) IterE (class in neuralkg_ind.model.RuleModel.IterE) IterELitModel (class in neuralkg_ind.lit_model.IterELitModel)

K

KBAT (class in neuralkg_ind.model.GNNModel.KBAT) KBAT_Loss (class in neuralkg_ind.loss.KBAT_Loss) KBATLitModel (class in neuralkg_ind.lit_model.KBATLitModel) KBATSampler (class in neuralkg_ind.data.Sampler) KGData (class in neuralkg_ind.data.DataPreprocess)

KGDataModule (class in neuralkg_ind.data.KGDataModule) KGDataset (class in neuralkg_ind.data.Sampler) KGEEvalData (class in neuralkg_ind.data.DataPreprocess) KGELitModel (class in neuralkg_ind.lit_model.KGELitModel) KGEModel (class in neuralkg_ind.model.GNNModel.MorsE) (neuralkg_ind.model.GNNModel.MorsE.MorsE attribute)

L

l2_loss() (neuralkg_ind.model.KGEModel.SimplE.SimplE method) label (neuralkg_ind.data.Sampler.CompGCNSampler attribute) (neuralkg_ind.data.Sampler.ConvSampler attribute) (neuralkg_ind.data.Sampler.GraphSampler attribute) (neuralkg_ind.data.Sampler.KBATSampler attribute) latent_dim (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE attribute) lef_mean (neuralkg_ind.data.Sampler.BernSampler attribute) link_predict() (in module neuralkg_ind.eval_task.link_prediction) link_predict_SEGNN() (in module neuralkg_ind.eval_task.link_prediction)

links2subgraphs() (in module neuralkg_ind.utils.tools) load_config() (in module neuralkg_ind.utils.tools) load_data_grail() (in module neuralkg_ind.utils.tools) (neuralkg_ind.data.DataPreprocess.GRData method) load_data_grail_ind() (neuralkg_ind.data.DataPreprocess.BaseGraph method) load_ind_data_grail() (in module neuralkg_ind.utils.tools) (neuralkg_ind.data.DataPreprocess.GRData method) log_metrics() (in module neuralkg_ind.utils.tools) log_step_metrics() (in module neuralkg_ind.utils.tools) Logsig_Loss (class in neuralkg_ind.loss.Logsig_Loss)

M

m_h2r (neuralkg_ind.data.DataPreprocess.GRData attribute) (neuralkg_ind.data.Sampler.TestSampler_hit attribute) m_t2r (neuralkg_ind.data.DataPreprocess.GRData attribute) (neuralkg_ind.data.Sampler.TestSampler_hit attribute) mapping_embedding() (neuralkg_ind.model.KGEModel.ComplEx.ComplEx method) (neuralkg_ind.model.KGEModel.DistMult.DistMult method) (neuralkg_ind.model.KGEModel.RotatE.RotatE method) (neuralkg_ind.model.KGEModel.TransE.TransE method) margin (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) (neuralkg_ind.model.KGEModel.DistMult.DistMult attribute) (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) (neuralkg_ind.model.KGEModel.PairRE.PairRE attribute) (neuralkg_ind.model.KGEModel.RotatE.RotatE attribute) (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) (neuralkg_ind.model.KGEModel.TransE.TransE attribute) (neuralkg_ind.model.KGEModel.TransH.TransH attribute) (neuralkg_ind.model.KGEModel.TransR.TransR attribute) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg_ind.model.RuleModel.IterE.IterE attribute) (neuralkg_ind.model.RuleModel.RugE.RugE attribute) Margin_Loss (class in neuralkg_ind.loss.Margin_Loss) max_neighbor (neuralkg_ind.data.Sampler.XTransESampler attribute) message() (neuralkg_ind.model.GNNModel.CompGCN.CompGCNCov method) (neuralkg_ind.model.GNNModel.layer.GNNLayer method) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv method) (neuralkg_ind.model.GNNModel.RGCN.RelGraphConv method) MetaGNNLitModel (class in neuralkg_ind.lit_model.MetaGNNLitModel) MetaSampler (class in neuralkg_ind.data.Sampler) MetaTrainGRData (class in neuralkg_ind.data.DataPreprocess) MetaValidGRData (class in neuralkg_ind.data.DataPreprocess) MLPAggregator (class in neuralkg_ind.model.GNNModel.layer) Model (class in neuralkg_ind.model.GNNModel.model) (class in neuralkg_ind.model.KGEModel.model) (class in neuralkg_ind.model.RuleModel.model) model (neuralkg_ind.loss.Adv_Loss.Adv_Loss attribute) (neuralkg_ind.loss.Softplus_Loss.Softplus_Loss attribute) model_name (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) module neuralkg_ind.data.base_data_module neuralkg_ind.data.DataPreprocess neuralkg_ind.data.Grounding neuralkg_ind.data.KGDataModule neuralkg_ind.data.RuleDataLoader neuralkg_ind.data.Sampler neuralkg_ind.eval_task.link_prediction neuralkg_ind.eval_task.triple_classification neuralkg_ind.lit_model.BaseLitModel neuralkg_ind.lit_model.CompGCNLitModel neuralkg_ind.lit_model.ConvELitModel neuralkg_ind.lit_model.CrossELitModel neuralkg_ind.lit_model.indGNNLitModel neuralkg_ind.lit_model.IterELitModel neuralkg_ind.lit_model.KBATLitModel neuralkg_ind.lit_model.KGELitModel neuralkg_ind.lit_model.MetaGNNLitModel neuralkg_ind.lit_model.RGCNLitModel neuralkg_ind.lit_model.RugELitModel neuralkg_ind.lit_model.SEGNNLitModel neuralkg_ind.lit_model.SNRILitModel neuralkg_ind.lit_model.XTransELitModel neuralkg_ind.loss.Adv_Loss neuralkg_ind.loss.ComplEx_NNE_AER_Loss neuralkg_ind.loss.Cross_Entropy_Loss neuralkg_ind.loss.CrossE_Loss neuralkg_ind.loss.KBAT_Loss neuralkg_ind.loss.Logsig_Loss neuralkg_ind.loss.Margin_Loss neuralkg_ind.loss.RGCN_Loss neuralkg_ind.loss.RugE_Loss neuralkg_ind.loss.SimplE_Loss neuralkg_ind.loss.Softplus_Loss neuralkg_ind.model.GNNModel.CompGCN neuralkg_ind.model.GNNModel.CoMPILE neuralkg_ind.model.GNNModel.Grail neuralkg_ind.model.GNNModel.KBAT neuralkg_ind.model.GNNModel.layer neuralkg_ind.model.GNNModel.model neuralkg_ind.model.GNNModel.MorsE neuralkg_ind.model.GNNModel.RGCN neuralkg_ind.model.GNNModel.RMPI neuralkg_ind.model.GNNModel.SEGNN neuralkg_ind.model.GNNModel.SNRI neuralkg_ind.model.GNNModel.XTransE neuralkg_ind.model.KGEModel.BoxE neuralkg_ind.model.KGEModel.ComplEx neuralkg_ind.model.KGEModel.ConvE neuralkg_ind.model.KGEModel.CrossE neuralkg_ind.model.KGEModel.DistMult neuralkg_ind.model.KGEModel.DualE neuralkg_ind.model.KGEModel.HAKE neuralkg_ind.model.KGEModel.model neuralkg_ind.model.KGEModel.PairRE neuralkg_ind.model.KGEModel.RotatE neuralkg_ind.model.KGEModel.SimplE neuralkg_ind.model.KGEModel.TransE neuralkg_ind.model.KGEModel.TransH neuralkg_ind.model.KGEModel.TransR neuralkg_ind.model.RuleModel.ComplEx_NNE_AER neuralkg_ind.model.RuleModel.IterE neuralkg_ind.model.RuleModel.model neuralkg_ind.model.RuleModel.RugE neuralkg_ind.utils.setup_parser neuralkg_ind.utils.tools

modules_weight (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) MorsE (class in neuralkg_ind.model.GNNModel.MorsE) MySpMM (class in neuralkg_ind.model.GNNModel.CoMPILE)

N

n_e (neuralkg_ind.model.GNNModel.SNRI.Discriminator attribute) n_g (neuralkg_ind.model.GNNModel.SNRI.Discriminator attribute) n_hop (neuralkg_ind.data.Sampler.KBATSampler attribute) neg_sample (neuralkg_ind.data.Sampler.XTransESampler attribute) nei_rel_path() (neuralkg_ind.model.GNNModel.SNRI.SNRI method) neighbours (neuralkg_ind.data.Sampler.KBATSampler attribute) neuralkg_ind.data.base_data_module module neuralkg_ind.data.DataPreprocess module neuralkg_ind.data.Grounding module neuralkg_ind.data.KGDataModule module neuralkg_ind.data.RuleDataLoader module neuralkg_ind.data.Sampler module neuralkg_ind.eval_task.link_prediction module neuralkg_ind.eval_task.triple_classification module neuralkg_ind.lit_model.BaseLitModel module neuralkg_ind.lit_model.CompGCNLitModel module neuralkg_ind.lit_model.ConvELitModel module neuralkg_ind.lit_model.CrossELitModel module neuralkg_ind.lit_model.indGNNLitModel module neuralkg_ind.lit_model.IterELitModel module neuralkg_ind.lit_model.KBATLitModel module neuralkg_ind.lit_model.KGELitModel module neuralkg_ind.lit_model.MetaGNNLitModel module neuralkg_ind.lit_model.RGCNLitModel module neuralkg_ind.lit_model.RugELitModel module neuralkg_ind.lit_model.SEGNNLitModel module neuralkg_ind.lit_model.SNRILitModel module neuralkg_ind.lit_model.XTransELitModel module neuralkg_ind.loss.Adv_Loss module neuralkg_ind.loss.ComplEx_NNE_AER_Loss module neuralkg_ind.loss.Cross_Entropy_Loss module neuralkg_ind.loss.CrossE_Loss module neuralkg_ind.loss.KBAT_Loss module neuralkg_ind.loss.Logsig_Loss module neuralkg_ind.loss.Margin_Loss module neuralkg_ind.loss.RGCN_Loss module neuralkg_ind.loss.RugE_Loss module neuralkg_ind.loss.SimplE_Loss module neuralkg_ind.loss.Softplus_Loss module neuralkg_ind.model.GNNModel.CompGCN module neuralkg_ind.model.GNNModel.CoMPILE module neuralkg_ind.model.GNNModel.Grail module neuralkg_ind.model.GNNModel.KBAT module

neuralkg_ind.model.GNNModel.layer module neuralkg_ind.model.GNNModel.model module neuralkg_ind.model.GNNModel.MorsE module neuralkg_ind.model.GNNModel.RGCN module neuralkg_ind.model.GNNModel.RMPI module neuralkg_ind.model.GNNModel.SEGNN module neuralkg_ind.model.GNNModel.SNRI module neuralkg_ind.model.GNNModel.XTransE module neuralkg_ind.model.KGEModel.BoxE module neuralkg_ind.model.KGEModel.ComplEx module neuralkg_ind.model.KGEModel.ConvE module neuralkg_ind.model.KGEModel.CrossE module neuralkg_ind.model.KGEModel.DistMult module neuralkg_ind.model.KGEModel.DualE module neuralkg_ind.model.KGEModel.HAKE module neuralkg_ind.model.KGEModel.model module neuralkg_ind.model.KGEModel.PairRE module neuralkg_ind.model.KGEModel.RotatE module neuralkg_ind.model.KGEModel.SimplE module neuralkg_ind.model.KGEModel.TransE module neuralkg_ind.model.KGEModel.TransH module neuralkg_ind.model.KGEModel.TransR module neuralkg_ind.model.RuleModel.ComplEx_NNE_AER module neuralkg_ind.model.RuleModel.IterE module neuralkg_ind.model.RuleModel.model module neuralkg_ind.model.RuleModel.RugE module neuralkg_ind.utils.setup_parser module neuralkg_ind.utils.tools module node_emb (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE attribute) node_label() (in module neuralkg_ind.utils.tools) node_norm_to_edge_norm() (neuralkg_ind.data.Sampler.CompGCNSampler method) (neuralkg_ind.data.Sampler.GraphSampler method) NodeLayer (class in neuralkg_ind.model.GNNModel.SEGNN) norm (neuralkg_ind.data.Sampler.CompGCNSampler attribute) (neuralkg_ind.data.Sampler.GraphSampler attribute) norm_vector (neuralkg_ind.model.KGEModel.TransH.TransH attribute) normalize() (neuralkg_ind.loss.Adv_Loss.Adv_Loss method) nrelation (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) num_bases (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) num_cand (neuralkg_ind.data.DataPreprocess.KGEEvalData attribute) num_ent (neuralkg_ind.data.DataPreprocess.KGEEvalData attribute) (neuralkg_ind.data.Sampler.CompGCNTestSampler attribute) (neuralkg_ind.data.Sampler.GraphTestSampler attribute) (neuralkg_ind.data.Sampler.TestSampler attribute) num_rels (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) num_workers (neuralkg_ind.data.RuleDataLoader.RuleDataLoader attribute)

O

out_dim (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute)

output_dim (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE attribute) override_config() (in module neuralkg_ind.utils.tools)

P

PairRE (class in neuralkg_ind.model.KGEModel.PairRE) phase_weight (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) pin_memory (neuralkg_ind.data.RuleDataLoader.RuleDataLoader attribute) precision (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel attribute) (neuralkg_ind.lit_model.IterELitModel.IterELitModel attribute) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg_ind.lit_model.KGELitModel.KGELitModel attribute) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel attribute) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg_ind.lit_model.RugELitModel.RugELitModel attribute) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel attribute) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel attribute) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel attribute) prefetch_factor (neuralkg_ind.data.RuleDataLoader.RuleDataLoader attribute) prepare_data() (neuralkg_ind.data.base_data_module.BaseDataModule method) (neuralkg_ind.data.KGDataModule.KGDataModule method) prepare_data_per_node (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel attribute) (neuralkg_ind.lit_model.IterELitModel.IterELitModel attribute) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg_ind.lit_model.KGELitModel.KGELitModel attribute) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel attribute) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg_ind.lit_model.RugELitModel.RugELitModel attribute) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel attribute) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel attribute) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel attribute)

prepare_features() (neuralkg_ind.data.Sampler.TestRMPISampler_hit method) (neuralkg_ind.data.Sampler.TestSampler_hit method) prepare_features_new() (neuralkg_ind.data.DataPreprocess.GRData method) prepare_subgraph() (neuralkg_ind.data.Sampler.TestRMPISampler_hit method) prepare_subgraphs() (neuralkg_ind.data.DataPreprocess.GRData method) propagate() (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv method) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv method) PropositionalizeRule() (neuralkg_ind.data.Grounding.GroundAllRules method)

Q

quaternion_init() (neuralkg_ind.model.KGEModel.DualE.DualE method)

R

readData() (neuralkg_ind.data.Grounding.GroundAllRules method) reduce_func() (neuralkg_ind.model.GNNModel.CompGCN.CompGCNCov method) reg_loss() (neuralkg_ind.loss.RGCN_Loss.RGCN_Loss method) regularize_loss() (neuralkg_ind.model.KGEModel.CrossE.CrossE method) reidx() (in module neuralkg_ind.utils.tools) reidx_withr() (in module neuralkg_ind.utils.tools) reidx_withr_ande() (in module neuralkg_ind.utils.tools) rel2id (neuralkg_ind.data.DataPreprocess.KGData attribute) rel_emb (neuralkg_ind.model.GNNModel.Grail.Grail attribute) (neuralkg_ind.model.GNNModel.RMPI.RMPI attribute) (neuralkg_ind.model.GNNModel.SNRI.SNRI attribute) (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute) (neuralkg_ind.model.KGEModel.DistMult.DistMult attribute) (neuralkg_ind.model.KGEModel.DualE.DualE attribute) (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) (neuralkg_ind.model.KGEModel.PairRE.PairRE attribute) (neuralkg_ind.model.KGEModel.RotatE.RotatE attribute) (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) (neuralkg_ind.model.KGEModel.TransE.TransE attribute) (neuralkg_ind.model.KGEModel.TransH.TransH attribute) (neuralkg_ind.model.KGEModel.TransR.TransR attribute) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg_ind.model.RuleModel.IterE.IterE attribute) (neuralkg_ind.model.RuleModel.RugE.RugE attribute) rel_head_emb (neuralkg_ind.model.GNNModel.MorsE.EntInit attribute) rel_inv_emb (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) rel_reverse_emb (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute) rel_tail_emb (neuralkg_ind.model.GNNModel.MorsE.EntInit attribute) relation (neuralkg_ind.data.Sampler.CompGCNSampler attribute) (neuralkg_ind.data.Sampler.GraphSampler attribute) Relation2Tuple (neuralkg_ind.data.DataPreprocess.KGData attribute)

relation_emb (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE attribute) relation_embedding (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) RelAttGraphConv (class in neuralkg_ind.model.GNNModel.Grail) RelCompGraphConv (class in neuralkg_ind.model.GNNModel.SNRI) RelGraphConv (class in neuralkg_ind.model.GNNModel.RGCN) RelMorsGraphConv (class in neuralkg_ind.model.GNNModel.MorsE) RelSub2Obj (neuralkg_ind.data.DataPreprocess.KGData attribute) remove_nodes() (in module neuralkg_ind.utils.tools) RevSampler (class in neuralkg_ind.data.DataPreprocess) RGCN (class in neuralkg_ind.model.GNNModel.Grail) (class in neuralkg_ind.model.GNNModel.MorsE) (class in neuralkg_ind.model.GNNModel.RGCN) rgcn (neuralkg_ind.model.GNNModel.MorsE.MorsE attribute) RGCN_Loss (class in neuralkg_ind.loss.RGCN_Loss) RGCNLitModel (class in neuralkg_ind.lit_model.RGCNLitModel) rig_mean (neuralkg_ind.data.Sampler.BernSampler attribute) RMPI (class in neuralkg_ind.model.GNNModel.RMPI) RMPISampler (class in neuralkg_ind.data.Sampler) RotatE (class in neuralkg_ind.model.KGEModel.RotatE) rt2h (neuralkg_ind.data.DataPreprocess.KGEEvalData attribute) rt2h_all (neuralkg_ind.data.Sampler.CompGCNTestSampler attribute) (neuralkg_ind.data.Sampler.GraphTestSampler attribute) (neuralkg_ind.data.Sampler.TestSampler attribute) rt2h_train (neuralkg_ind.data.DataPreprocess.KGData attribute) (neuralkg_ind.data.DataPreprocess.RevSampler attribute) RugE (class in neuralkg_ind.model.RuleModel.RugE) RugE_Loss (class in neuralkg_ind.loss.RugE_Loss) RugELitModel (class in neuralkg_ind.lit_model.RugELitModel) RuleDataLoader (class in neuralkg_ind.data.RuleDataLoader) RuleDataset (class in neuralkg_ind.data.RuleDataLoader) run_axiom_probability() (neuralkg_ind.model.RuleModel.IterE.IterE method)

S

s_bias1 (neuralkg_ind.model.GNNModel.SNRI.Discriminator attribute) s_bias2 (neuralkg_ind.model.GNNModel.SNRI.Discriminator attribute) sam_negative() (neuralkg_ind.data.Sampler.KBATSampler method) sample_neg() (in module neuralkg_ind.utils.tools) sample_one_subgraph() (in module neuralkg_ind.utils.tools) sampler (neuralkg_ind.data.RuleDataLoader.RuleDataLoader attribute) (neuralkg_ind.data.Sampler.CompGCNTestSampler attribute) (neuralkg_ind.data.Sampler.GraphTestSampler attribute) (neuralkg_ind.data.Sampler.TestRMPISampler_auc attribute) (neuralkg_ind.data.Sampler.TestRMPISampler_hit attribute) (neuralkg_ind.data.Sampler.TestSampler attribute) (neuralkg_ind.data.Sampler.TestSampler_auc attribute) (neuralkg_ind.data.Sampler.TestSampler_hit attribute) (neuralkg_ind.data.Sampler.ValidRMPISampler attribute) (neuralkg_ind.data.Sampler.ValidSampler attribute) sampling() (neuralkg_ind.data.Sampler.AdvSampler method) (neuralkg_ind.data.Sampler.AllSampler method) (neuralkg_ind.data.Sampler.BernSampler method) (neuralkg_ind.data.Sampler.CompGCNSampler method) (neuralkg_ind.data.Sampler.CompGCNTestSampler method) (neuralkg_ind.data.Sampler.ConvSampler method) (neuralkg_ind.data.Sampler.CrossESampler method) (neuralkg_ind.data.Sampler.GraphSampler method) (neuralkg_ind.data.Sampler.GraphTestSampler method) (neuralkg_ind.data.Sampler.KBATSampler method) (neuralkg_ind.data.Sampler.MetaSampler method) (neuralkg_ind.data.Sampler.RMPISampler method) (neuralkg_ind.data.Sampler.SEGNNTestSampler method) (neuralkg_ind.data.Sampler.SEGNNTrainSampler method) (neuralkg_ind.data.Sampler.SubSampler method) (neuralkg_ind.data.Sampler.TestMetaSampler_auc method) (neuralkg_ind.data.Sampler.TestMetaSampler_hit method) (neuralkg_ind.data.Sampler.TestRMPISampler_auc method) (neuralkg_ind.data.Sampler.TestRMPISampler_hit method) (neuralkg_ind.data.Sampler.TestSampler method) (neuralkg_ind.data.Sampler.TestSampler_auc method) (neuralkg_ind.data.Sampler.TestSampler_hit method) (neuralkg_ind.data.Sampler.UniSampler method) (neuralkg_ind.data.Sampler.ValidMetaSampler method) (neuralkg_ind.data.Sampler.ValidRMPISampler method) (neuralkg_ind.data.Sampler.ValidSampler method) (neuralkg_ind.data.Sampler.XTransESampler method) sampling_keys() (neuralkg_ind.data.Sampler.AllSampler method) (neuralkg_ind.data.Sampler.BernSampler static method) (neuralkg_ind.data.Sampler.ConvSampler method) (neuralkg_ind.data.Sampler.CrossESampler method) sampling_negative() (neuralkg_ind.data.Sampler.GraphSampler method) (neuralkg_ind.data.Sampler.KBATSampler method) sampling_positive() (neuralkg_ind.data.Sampler.GraphSampler method) save_config() (in module neuralkg_ind.utils.tools) score_embedding() (neuralkg_ind.model.KGEModel.ComplEx.ComplEx method) (neuralkg_ind.model.KGEModel.DistMult.DistMult method) (neuralkg_ind.model.KGEModel.RotatE.RotatE method) (neuralkg_ind.model.KGEModel.TransE.TransE method)

score_func() (neuralkg_ind.model.GNNModel.XTransE.XTransE method) (neuralkg_ind.model.KGEModel.BoxE.BoxE method) (neuralkg_ind.model.KGEModel.ComplEx.ComplEx method) (neuralkg_ind.model.KGEModel.ConvE.ConvE method) (neuralkg_ind.model.KGEModel.CrossE.CrossE method) (neuralkg_ind.model.KGEModel.DistMult.DistMult method) (neuralkg_ind.model.KGEModel.DualE.DualE method) (neuralkg_ind.model.KGEModel.HAKE.HAKE method) (neuralkg_ind.model.KGEModel.model.Model method) (neuralkg_ind.model.KGEModel.PairRE.PairRE method) (neuralkg_ind.model.KGEModel.RotatE.RotatE method) (neuralkg_ind.model.KGEModel.SimplE.SimplE method) (neuralkg_ind.model.KGEModel.TransE.TransE method) (neuralkg_ind.model.KGEModel.TransH.TransH method) (neuralkg_ind.model.KGEModel.TransR.TransR method) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg_ind.model.RuleModel.IterE.IterE method) (neuralkg_ind.model.RuleModel.model.Model method) (neuralkg_ind.model.RuleModel.RugE.RugE method) SEGNN (class in neuralkg_ind.model.GNNModel.SEGNN) SEGNNLitModel (class in neuralkg_ind.lit_model.SEGNNLitModel) SEGNNTestSampler (class in neuralkg_ind.data.Sampler) SEGNNTrainProcess (class in neuralkg_ind.data.Sampler) SEGNNTrainSampler (class in neuralkg_ind.data.Sampler) self_loop_weight (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) serialize() (in module neuralkg_ind.utils.tools) set_logger() (in module neuralkg_ind.utils.tools) setup() (neuralkg_ind.data.base_data_module.BaseDataModule method) (neuralkg_ind.data.KGDataModule.KGDataModule method) setup_parser() (in module neuralkg_ind.utils.setup_parser) sim() (neuralkg_ind.model.RuleModel.IterE.IterE method) SimplE (class in neuralkg_ind.model.KGEModel.SimplE) SimplE_Loss (class in neuralkg_ind.loss.SimplE_Loss) SNRI (class in neuralkg_ind.model.GNNModel.SNRI) SNRILitModel (class in neuralkg_ind.lit_model.SNRILitModel) Softplus_Loss (class in neuralkg_ind.loss.Softplus_Loss) sp_row_vec_from_idx_list() (in module neuralkg_ind.utils.tools) sparse_dense_mul() (neuralkg_ind.model.GNNModel.RMPI.RMPI static method) sparse_index_select() (neuralkg_ind.model.GNNModel.RMPI.RMPI static method) SpecialSpmmFinal (class in neuralkg_ind.model.GNNModel.KBAT) SpecialSpmmFunctionFinal (class in neuralkg_ind.model.GNNModel.KBAT) split_embedding() (neuralkg_ind.model.RuleModel.IterE.IterE method) ssp_graph (neuralkg_ind.data.DataPreprocess.GRData attribute) ssp_multigraph_to_dgl() (in module neuralkg_ind.utils.tools) subgraph_extraction_labeling() (in module neuralkg_ind.utils.tools) subgraphs_db (neuralkg_ind.data.DataPreprocess.MetaTrainGRData attribute) (neuralkg_ind.data.DataPreprocess.MetaValidGRData attribute) SubSampler (class in neuralkg_ind.data.Sampler) SUMAggregator (class in neuralkg_ind.model.GNNModel.layer)

T

t2rh_train (neuralkg_ind.data.DataPreprocess.KGData attribute) t_weighted_vector (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute) tail_batch() (neuralkg_ind.data.DataPreprocess.BaseSampler method) (neuralkg_ind.data.DataPreprocess.RevSampler method) tail_predict() (in module neuralkg_ind.eval_task.link_prediction) tail_predict_SEGNN() (in module neuralkg_ind.eval_task.link_prediction) test_dataloader() (neuralkg_ind.data.base_data_module.BaseDataModule method) (neuralkg_ind.data.KGDataModule.KGDataModule method) test_epoch_end() (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel method) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel method) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel method) (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel method) (neuralkg_ind.lit_model.KGELitModel.KGELitModel method) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel method) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg_ind.lit_model.RugELitModel.RugELitModel method) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel method) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel method) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel method) test_step() (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel method) (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel method) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel method) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel method) (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel method) (neuralkg_ind.lit_model.KGELitModel.KGELitModel method) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel method) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg_ind.lit_model.RugELitModel.RugELitModel method) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel method) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel method) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel method) test_triples (neuralkg_ind.data.DataPreprocess.KGData attribute) TestMetaSampler_auc (class in neuralkg_ind.data.Sampler) TestMetaSampler_hit (class in neuralkg_ind.data.Sampler) TestRMPISampler_auc (class in neuralkg_ind.data.Sampler) TestRMPISampler_hit (class in neuralkg_ind.data.Sampler) TestSampler (class in neuralkg_ind.data.Sampler) TestSampler_auc (class in neuralkg_ind.data.Sampler) TestSampler_hit (class in neuralkg_ind.data.Sampler) timeout (neuralkg_ind.data.RuleDataLoader.RuleDataLoader attribute) tr_bias (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute) train_dataloader() (neuralkg_ind.data.base_data_module.BaseDataModule method) (neuralkg_ind.data.KGDataModule.KGDataModule method) train_triples (neuralkg_ind.data.DataPreprocess.KGData attribute) training (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel attribute) (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel attribute) (neuralkg_ind.lit_model.IterELitModel.IterELitModel attribute) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg_ind.lit_model.KGELitModel.KGELitModel attribute) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel attribute) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg_ind.lit_model.RugELitModel.RugELitModel attribute) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel attribute) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel attribute) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel attribute) (neuralkg_ind.loss.Adv_Loss.Adv_Loss attribute) (neuralkg_ind.loss.ComplEx_NNE_AER_Loss.ComplEx_NNE_AER_Loss attribute) (neuralkg_ind.loss.Cross_Entropy_Loss.Cross_Entropy_Loss attribute) (neuralkg_ind.loss.CrossE_Loss.CrossE_Loss attribute) (neuralkg_ind.loss.KBAT_Loss.KBAT_Loss attribute) (neuralkg_ind.loss.Logsig_Loss.Logsig_Loss attribute) (neuralkg_ind.loss.Margin_Loss.Margin_Loss attribute) (neuralkg_ind.loss.RGCN_Loss.RGCN_Loss attribute) (neuralkg_ind.loss.RugE_Loss.RugE_Loss attribute) (neuralkg_ind.loss.SimplE_Loss.SimplE_Loss attribute) (neuralkg_ind.loss.Softplus_Loss.Softplus_Loss attribute) (neuralkg_ind.model.GNNModel.CompGCN.CompGCN attribute) (neuralkg_ind.model.GNNModel.CompGCN.CompGCNCov attribute) (neuralkg_ind.model.GNNModel.CoMPILE.CoMPILE attribute) (neuralkg_ind.model.GNNModel.Grail.Grail attribute) (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.Grail.RGCN attribute) (neuralkg_ind.model.GNNModel.KBAT.GraphAttentionLayer attribute) (neuralkg_ind.model.GNNModel.KBAT.KBAT attribute) (neuralkg_ind.model.GNNModel.KBAT.SpecialSpmmFinal attribute) (neuralkg_ind.model.GNNModel.layer.Aggregator attribute) (neuralkg_ind.model.GNNModel.layer.BatchGRU attribute) (neuralkg_ind.model.GNNModel.layer.GNNLayer attribute) (neuralkg_ind.model.GNNModel.layer.GRUAggregator attribute) (neuralkg_ind.model.GNNModel.layer.MLPAggregator attribute) (neuralkg_ind.model.GNNModel.layer.SUMAggregator attribute) (neuralkg_ind.model.GNNModel.model.Model attribute) (neuralkg_ind.model.GNNModel.MorsE.EntInit attribute) (neuralkg_ind.model.GNNModel.MorsE.KGEModel attribute) (neuralkg_ind.model.GNNModel.MorsE.MorsE attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RGCN attribute) (neuralkg_ind.model.GNNModel.RGCN.RelGraphConv attribute) (neuralkg_ind.model.GNNModel.RGCN.RGCN attribute) (neuralkg_ind.model.GNNModel.RMPI.RMPI attribute) (neuralkg_ind.model.GNNModel.SEGNN.CompLayer attribute) (neuralkg_ind.model.GNNModel.SEGNN.EdgeLayer attribute) (neuralkg_ind.model.GNNModel.SEGNN.NodeLayer attribute) (neuralkg_ind.model.GNNModel.SEGNN.SEGNN attribute) (neuralkg_ind.model.GNNModel.SNRI.Discriminator attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.SNRI attribute) (neuralkg_ind.model.GNNModel.XTransE.XTransE attribute) (neuralkg_ind.model.KGEModel.BoxE.BoxE attribute) (neuralkg_ind.model.KGEModel.ComplEx.ComplEx attribute) (neuralkg_ind.model.KGEModel.ConvE.ConvE attribute) (neuralkg_ind.model.KGEModel.CrossE.CrossE attribute) (neuralkg_ind.model.KGEModel.DistMult.DistMult attribute) (neuralkg_ind.model.KGEModel.DualE.DualE attribute) (neuralkg_ind.model.KGEModel.HAKE.HAKE attribute) (neuralkg_ind.model.KGEModel.model.Model attribute) (neuralkg_ind.model.KGEModel.PairRE.PairRE attribute) (neuralkg_ind.model.KGEModel.RotatE.RotatE attribute) (neuralkg_ind.model.KGEModel.SimplE.SimplE attribute) (neuralkg_ind.model.KGEModel.TransE.TransE attribute) (neuralkg_ind.model.KGEModel.TransH.TransH attribute) (neuralkg_ind.model.KGEModel.TransR.TransR attribute) (neuralkg_ind.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg_ind.model.RuleModel.IterE.IterE attribute) (neuralkg_ind.model.RuleModel.model.Model attribute) (neuralkg_ind.model.RuleModel.RugE.RugE attribute)

training_epoch_end() (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) (neuralkg_ind.lit_model.RugELitModel.RugELitModel method) training_step() (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel method) (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel method) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel method) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel method) (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel method) (neuralkg_ind.lit_model.KGELitModel.KGELitModel method) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel method) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg_ind.lit_model.RugELitModel.RugELitModel method) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel method) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel method) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel method) TrainTriples (neuralkg_ind.data.DataPreprocess.KGData attribute) TransE (class in neuralkg_ind.model.KGEModel.TransE) transe_func() (neuralkg_ind.model.GNNModel.XTransE.XTransE method) transfer_matrix (neuralkg_ind.model.KGEModel.TransR.TransR attribute) TransH (class in neuralkg_ind.model.KGEModel.TransH) TransR (class in neuralkg_ind.model.KGEModel.TransR) tri2emb() (neuralkg_ind.model.GNNModel.RGCN.RGCN method) (neuralkg_ind.model.KGEModel.model.Model method) (neuralkg_ind.model.RuleModel.model.Model method) triples (neuralkg_ind.data.DataPreprocess.KGEEvalData attribute) (neuralkg_ind.data.Sampler.CompGCNSampler attribute) (neuralkg_ind.data.Sampler.CompGCNTestSampler attribute) (neuralkg_ind.data.Sampler.ConvSampler attribute) (neuralkg_ind.data.Sampler.GraphSampler attribute) (neuralkg_ind.data.Sampler.GraphTestSampler attribute) (neuralkg_ind.data.Sampler.KBATSampler attribute) (neuralkg_ind.data.Sampler.XTransESampler attribute) triples_Con (neuralkg_ind.data.Sampler.KBATSampler attribute) triples_GAT_neg (neuralkg_ind.data.Sampler.KBATSampler attribute) triples_GAT_pos (neuralkg_ind.data.Sampler.KBATSampler attribute)

U

uni_sampling() (neuralkg_ind.data.Sampler.UniSampler method) UniSampler (class in neuralkg_ind.data.Sampler) update_axiom() (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) update_embedding() (neuralkg_ind.model.GNNModel.layer.Aggregator method) (neuralkg_ind.model.GNNModel.layer.GRUAggregator method) (neuralkg_ind.model.GNNModel.layer.MLPAggregator method) (neuralkg_ind.model.GNNModel.layer.SUMAggregator method) update_train_triples() (neuralkg_ind.model.RuleModel.IterE.IterE method) update_valid_axioms() (neuralkg_ind.model.RuleModel.IterE.IterE method) use_amp (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel attribute) (neuralkg_ind.lit_model.IterELitModel.IterELitModel attribute) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg_ind.lit_model.KGELitModel.KGELitModel attribute) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel attribute) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg_ind.lit_model.RugELitModel.RugELitModel attribute) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel attribute) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel attribute) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel attribute)

V

val_dataloader() (neuralkg_ind.data.base_data_module.BaseDataModule method) (neuralkg_ind.data.KGDataModule.KGDataModule method) valid_triples (neuralkg_ind.data.DataPreprocess.KGData attribute) validation_epoch_end() (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel method) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel method) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel method) (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel method) (neuralkg_ind.lit_model.KGELitModel.KGELitModel method) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel method) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg_ind.lit_model.RugELitModel.RugELitModel method) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel method) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel method) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel method)

validation_step() (neuralkg_ind.lit_model.BaseLitModel.BaseLitModel method) (neuralkg_ind.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg_ind.lit_model.ConvELitModel.ConvELitModel method) (neuralkg_ind.lit_model.CrossELitModel.CrossELitModel method) (neuralkg_ind.lit_model.indGNNLitModel.indGNNLitModel method) (neuralkg_ind.lit_model.IterELitModel.IterELitModel method) (neuralkg_ind.lit_model.KBATLitModel.KBATLitModel method) (neuralkg_ind.lit_model.KGELitModel.KGELitModel method) (neuralkg_ind.lit_model.MetaGNNLitModel.MetaGNNLitModel method) (neuralkg_ind.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg_ind.lit_model.RugELitModel.RugELitModel method) (neuralkg_ind.lit_model.SEGNNLitModel.SEGNNLitModel method) (neuralkg_ind.lit_model.SNRILitModel.SNRILitModel method) (neuralkg_ind.lit_model.XTransELitModel.XTransELitModel method) ValidMetaSampler (class in neuralkg_ind.data.Sampler) ValidRMPISampler (class in neuralkg_ind.data.Sampler) ValidSampler (class in neuralkg_ind.data.Sampler)

W

w_comp (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) w_rel2ent (neuralkg_ind.model.GNNModel.SNRI.SNRI attribute)

weight (neuralkg_ind.model.GNNModel.Grail.RelAttGraphConv attribute) (neuralkg_ind.model.GNNModel.MorsE.RelMorsGraphConv attribute) (neuralkg_ind.model.GNNModel.SNRI.RelCompGraphConv attribute) weights_init() (neuralkg_ind.model.GNNModel.SNRI.Discriminator method)

X

XTransE (class in neuralkg_ind.model.GNNModel.XTransE)

XTransELitModel (class in neuralkg_ind.lit_model.XTransELitModel) XTransESampler (class in neuralkg_ind.data.Sampler)