Index

A

add_reverse_relation() (neuralkg.data.DataPreprocess.RevSampler method) add_reverse_triples() (neuralkg.data.DataPreprocess.RevSampler method) add_to_argparse() (neuralkg.data.base_data_module.BaseDataModule static method) (neuralkg.lit_model.BaseLitModel.BaseLitModel static method) (neuralkg.lit_model.KGELitModel.KGELitModel static method) adj_matrix (neuralkg.data.Sampler.KBATSampler attribute) Adv_Loss (class in neuralkg.loss.Adv_Loss) AdvSampler (class in neuralkg.data.Sampler) all_true_triples (neuralkg.data.DataPreprocess.KGData attribute) allow_zero_length_dataloader_with_multiple_devices (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg.lit_model.IterELitModel.IterELitModel attribute) (neuralkg.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg.lit_model.RugELitModel.RugELitModel attribute) (neuralkg.lit_model.XTransELitModel.XTransELitModel attribute) AllSampler (class in neuralkg.data.Sampler)

args (neuralkg.data.DataPreprocess.KGData attribute) (neuralkg.loss.Adv_Loss.Adv_Loss attribute) (neuralkg.model.GNNModel.CompGCN.CompGCN attribute) (neuralkg.model.GNNModel.KBAT.KBAT attribute) (neuralkg.model.GNNModel.RGCN.RGCN attribute) (neuralkg.model.GNNModel.XTransE.XTransE attribute) (neuralkg.model.KGEModel.BoxE.BoxE attribute) (neuralkg.model.KGEModel.ConvE.ConvE attribute) (neuralkg.model.KGEModel.CrossE.CrossE attribute) (neuralkg.model.KGEModel.DistMult.DistMult attribute) (neuralkg.model.KGEModel.RotatE.RotatE attribute) (neuralkg.model.KGEModel.SimplE.SimplE attribute) (neuralkg.model.KGEModel.TransE.TransE attribute) (neuralkg.model.KGEModel.TransH.TransH attribute) (neuralkg.model.KGEModel.TransR.TransR attribute) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg.model.RuleModel.IterE.IterE attribute) (neuralkg.model.RuleModel.RugE.RugE attribute)

B

backward() (neuralkg.model.GNNModel.KBAT.SpecialSpmmFunctionFinal static method) BaseDataModule (class in neuralkg.data.base_data_module) BaseLitModel (class in neuralkg.lit_model.BaseLitModel) BaseSampler (class in neuralkg.data.DataPreprocess) batch_size (neuralkg.data.RuleDataLoader.RuleDataLoader attribute)

BernSampler (class in neuralkg.data.Sampler) bfs() (neuralkg.data.Sampler.KBATSampler method) BoxE (class in neuralkg.model.KGEModel.BoxE) build_graph() (neuralkg.data.Sampler.GraphSampler method) build_hidden_layer() (neuralkg.model.GNNModel.RGCN.RGCN method) build_model() (neuralkg.model.GNNModel.RGCN.RGCN method)

C

cal_Con_score() (neuralkg.model.GNNModel.KBAT.KBAT method) cal_GAT_score() (neuralkg.model.GNNModel.KBAT.KBAT method) calc_bern() (neuralkg.data.Sampler.BernSampler method) calc_freq() (neuralkg.data.Sampler.AdvSampler method) calc_ranks() (in module neuralkg.eval_task.RGCN_prediction) calc_score() (neuralkg.model.KGEModel.BoxE.BoxE method) collect_results() (neuralkg.lit_model.BaseLitModel.BaseLitModel method) comp() (neuralkg.model.GNNModel.CompGCN.CompGCNCov method) comp_deg_norm() (neuralkg.data.Sampler.GraphSampler method) CompGCN (class in neuralkg.model.GNNModel.CompGCN) CompGCNCov (class in neuralkg.model.GNNModel.CompGCN) CompGCNLitModel (class in neuralkg.lit_model.CompGCNLitModel) CompGCNSampler (class in neuralkg.data.Sampler) CompGCNTestSampler (class in neuralkg.data.Sampler) ComplEx (class in neuralkg.model.KGEModel.ComplEx) ComplEx_NNE_AER (class in neuralkg.model.RuleModel.ComplEx_NNE_AER) ComplEx_NNE_AER_Loss (class in neuralkg.loss.ComplEx_NNE_AER_Loss) concat() (neuralkg.model.GNNModel.CompGCN.CompGCN method) Config (class in neuralkg.data.base_data_module) (class in neuralkg.lit_model.BaseLitModel) configure_optimizers() (neuralkg.lit_model.BaseLitModel.BaseLitModel method) (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg.lit_model.ConvELitModel.ConvELitModel method) (neuralkg.lit_model.CrossELitModel.CrossELitModel method) (neuralkg.lit_model.IterELitModel.IterELitModel method) (neuralkg.lit_model.KBATLitModel.KBATLitModel method) (neuralkg.lit_model.KGELitModel.KGELitModel method) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg.lit_model.RugELitModel.RugELitModel method) (neuralkg.lit_model.XTransELitModel.XTransELitModel method)

ConvE (class in neuralkg.model.KGEModel.ConvE) ConvE() (neuralkg.model.GNNModel.CompGCN.CompGCN method) ConvELitModel (class in neuralkg.lit_model.ConvELitModel) ConvSampler (class in neuralkg.data.Sampler) corrupt_head() (neuralkg.data.DataPreprocess.BaseSampler method) (neuralkg.data.DataPreprocess.RevSampler method) corrupt_tail() (neuralkg.data.DataPreprocess.BaseSampler method) (neuralkg.data.DataPreprocess.RevSampler method) count_frequency() (neuralkg.data.DataPreprocess.KGData static method) Cross_Entropy_Loss (class in neuralkg.loss.Cross_Entropy_Loss) cross_sampling_flag (neuralkg.data.Sampler.UniSampler attribute) CrossE (class in neuralkg.model.KGEModel.CrossE) CrossE_Loss (class in neuralkg.loss.CrossE_Loss) CrossELitModel (class in neuralkg.lit_model.CrossELitModel) CrossESampler (class in neuralkg.data.Sampler)

D

dataset (neuralkg.data.RuleDataLoader.RuleDataLoader attribute) dist() (neuralkg.model.KGEModel.BoxE.BoxE method)

DistMult (class in neuralkg.model.KGEModel.DistMult) DistMult() (neuralkg.model.GNNModel.CompGCN.CompGCN method) drop_last (neuralkg.data.RuleDataLoader.RuleDataLoader attribute)

E

embedding_range (neuralkg.model.KGEModel.DistMult.DistMult attribute) (neuralkg.model.KGEModel.RotatE.RotatE attribute) (neuralkg.model.KGEModel.SimplE.SimplE attribute) (neuralkg.model.KGEModel.TransE.TransE attribute) (neuralkg.model.KGEModel.TransH.TransH attribute) (neuralkg.model.KGEModel.TransR.TransR attribute) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg.model.RuleModel.IterE.IterE attribute) (neuralkg.model.RuleModel.RugE.RugE attribute) ent2id (neuralkg.data.DataPreprocess.KGData attribute) ent_emb (neuralkg.model.KGEModel.CrossE.CrossE attribute) (neuralkg.model.KGEModel.DistMult.DistMult attribute) (neuralkg.model.KGEModel.RotatE.RotatE attribute) (neuralkg.model.KGEModel.TransE.TransE attribute) (neuralkg.model.KGEModel.TransH.TransH attribute) (neuralkg.model.KGEModel.TransR.TransR attribute) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg.model.RuleModel.IterE.IterE attribute) (neuralkg.model.RuleModel.RugE.RugE attribute)

ent_h_emb (neuralkg.model.KGEModel.SimplE.SimplE attribute) ent_t_emb (neuralkg.model.KGEModel.SimplE.SimplE attribute) entity (neuralkg.data.Sampler.GraphSampler attribute) epsilon (neuralkg.model.KGEModel.DistMult.DistMult attribute) (neuralkg.model.KGEModel.RotatE.RotatE attribute) (neuralkg.model.KGEModel.SimplE.SimplE attribute) (neuralkg.model.KGEModel.TransE.TransE attribute) (neuralkg.model.KGEModel.TransH.TransH attribute) (neuralkg.model.KGEModel.TransR.TransR attribute) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg.model.RuleModel.IterE.IterE attribute) (neuralkg.model.RuleModel.RugE.RugE attribute)

F

forward() (neuralkg.lit_model.BaseLitModel.BaseLitModel method) (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg.lit_model.ConvELitModel.ConvELitModel method) (neuralkg.lit_model.CrossELitModel.CrossELitModel method) (neuralkg.lit_model.IterELitModel.IterELitModel method) (neuralkg.lit_model.KBATLitModel.KBATLitModel method) (neuralkg.lit_model.KGELitModel.KGELitModel method) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg.lit_model.RugELitModel.RugELitModel method) (neuralkg.lit_model.XTransELitModel.XTransELitModel method) (neuralkg.loss.Adv_Loss.Adv_Loss method) (neuralkg.loss.ComplEx_NNE_AER_Loss.ComplEx_NNE_AER_Loss method) (neuralkg.loss.Cross_Entropy_Loss.Cross_Entropy_Loss method) (neuralkg.loss.CrossE_Loss.CrossE_Loss method) (neuralkg.loss.IterE_Loss.IterE_Loss method) (neuralkg.loss.KBAT_Loss.KBAT_Loss method) (neuralkg.loss.Margin_Loss.Margin_Loss method) (neuralkg.loss.RGCN_Loss.RGCN_Loss method) (neuralkg.loss.RugE_Loss.RugE_Loss method) (neuralkg.loss.SimplE_Loss.SimplE_Loss method) (neuralkg.model.GNNModel.CompGCN.CompGCN method) (neuralkg.model.GNNModel.CompGCN.CompGCNCov method) (neuralkg.model.GNNModel.KBAT.GraphAttentionLayer method) (neuralkg.model.GNNModel.KBAT.KBAT method) (neuralkg.model.GNNModel.KBAT.SpecialSpmmFinal method) (neuralkg.model.GNNModel.KBAT.SpecialSpmmFunctionFinal static method) (neuralkg.model.GNNModel.RGCN.RGCN method) (neuralkg.model.GNNModel.XTransE.XTransE method) (neuralkg.model.KGEModel.BoxE.BoxE method) (neuralkg.model.KGEModel.ComplEx.ComplEx method) (neuralkg.model.KGEModel.ConvE.ConvE method) (neuralkg.model.KGEModel.CrossE.CrossE method) (neuralkg.model.KGEModel.DistMult.DistMult method) (neuralkg.model.KGEModel.model.Model method) (neuralkg.model.KGEModel.RotatE.RotatE method) (neuralkg.model.KGEModel.SimplE.SimplE method) (neuralkg.model.KGEModel.TransE.TransE method) (neuralkg.model.KGEModel.TransH.TransH method) (neuralkg.model.KGEModel.TransR.TransR method) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg.model.RuleModel.IterE.IterE method) (neuralkg.model.RuleModel.RugE.RugE method)

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

G

generate_new_train_triples() (neuralkg.model.RuleModel.IterE.IterE method) get_all_true_triples() (neuralkg.data.DataPreprocess.BaseSampler method) (neuralkg.data.DataPreprocess.RevSampler method) get_axiom() (neuralkg.model.RuleModel.IterE.IterE method) get_batch_nhop_neighbors_all() (neuralkg.data.Sampler.KBATSampler method) get_config() (neuralkg.data.base_data_module.BaseDataModule method) get_data_config() (neuralkg.data.KGDataModule.KGDataModule method) get_emb() (neuralkg.model.KGEModel.SimplE.SimplE method) get_h2rt_t2hr_from_train() (neuralkg.data.DataPreprocess.KGData method) get_hr2t_rt2h_from_all() (neuralkg.data.Sampler.CompGCNTestSampler method) (neuralkg.data.Sampler.GraphTestSampler method) (neuralkg.data.Sampler.TestSampler method) get_hr2t_rt2h_from_train() (neuralkg.data.DataPreprocess.KGData method) get_hr_trian() (neuralkg.data.DataPreprocess.KGData method) get_id() (neuralkg.data.DataPreprocess.KGData method) get_neighbors() (neuralkg.data.Sampler.KBATSampler method) get_param() (neuralkg.model.GNNModel.CompGCN.CompGCNCov method) get_rule() (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg.model.RuleModel.IterE.IterE method) get_sampling_keys() (neuralkg.data.Sampler.CompGCNSampler method) (neuralkg.data.Sampler.CompGCNTestSampler method) (neuralkg.data.Sampler.GraphSampler method) (neuralkg.data.Sampler.GraphTestSampler method) (neuralkg.data.Sampler.KBATSampler method) (neuralkg.data.Sampler.TestSampler method) (neuralkg.data.Sampler.UniSampler method) (neuralkg.data.Sampler.XTransESampler method) get_score() (neuralkg.model.GNNModel.CompGCN.CompGCN method) (neuralkg.model.GNNModel.RGCN.RGCN method) (neuralkg.model.GNNModel.XTransE.XTransE method) (neuralkg.model.KGEModel.BoxE.BoxE method) (neuralkg.model.KGEModel.ComplEx.ComplEx method) (neuralkg.model.KGEModel.ConvE.ConvE method) (neuralkg.model.KGEModel.CrossE.CrossE method) (neuralkg.model.KGEModel.DistMult.DistMult method) (neuralkg.model.KGEModel.RotatE.RotatE method) (neuralkg.model.KGEModel.SimplE.SimplE method) (neuralkg.model.KGEModel.TransE.TransE method) (neuralkg.model.KGEModel.TransH.TransH method) (neuralkg.model.KGEModel.TransR.TransR method) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg.model.RuleModel.IterE.IterE method) (neuralkg.model.RuleModel.RugE.RugE method)

get_test() (neuralkg.data.DataPreprocess.BaseSampler method) (neuralkg.data.DataPreprocess.RevSampler method) get_train() (neuralkg.data.DataPreprocess.BaseSampler method) (neuralkg.data.DataPreprocess.RevSampler method) get_train_bs() (neuralkg.data.KGDataModule.KGDataModule method) get_triples_id() (neuralkg.data.DataPreprocess.KGData method) get_unique_entity() (neuralkg.data.Sampler.KBATSampler method) get_valid() (neuralkg.data.DataPreprocess.BaseSampler method) (neuralkg.data.DataPreprocess.RevSampler method) graph (neuralkg.data.Sampler.CompGCNSampler attribute) (neuralkg.data.Sampler.GraphSampler attribute) (neuralkg.data.Sampler.KBATSampler attribute) GraphAttentionLayer (class in neuralkg.model.GNNModel.KBAT) GraphSampler (class in neuralkg.data.Sampler) GraphTestSampler (class in neuralkg.data.Sampler) GroundAllRules (class in neuralkg.data.Grounding) groundRule() (neuralkg.data.Grounding.GroundAllRules method)

H

h2rt_train (neuralkg.data.DataPreprocess.KGData attribute) h_mask (neuralkg.data.Sampler.XTransESampler attribute) h_neighbor (neuralkg.data.Sampler.XTransESampler attribute) h_weighted_vector (neuralkg.model.KGEModel.CrossE.CrossE attribute) head_batch() (neuralkg.data.DataPreprocess.BaseSampler method) (neuralkg.data.DataPreprocess.RevSampler method)

head_predict() (in module neuralkg.eval_task.RGCN_prediction) hr2t_all (neuralkg.data.Sampler.CompGCNTestSampler attribute) (neuralkg.data.Sampler.GraphTestSampler attribute) (neuralkg.data.Sampler.TestSampler attribute) hr2t_train (neuralkg.data.DataPreprocess.KGData attribute) (neuralkg.data.DataPreprocess.RevSampler attribute) hr_bias (neuralkg.model.KGEModel.CrossE.CrossE attribute)

I

id2ent (neuralkg.data.DataPreprocess.KGData attribute) id2rel (neuralkg.data.DataPreprocess.KGData attribute) init_ConvKB_emb() (neuralkg.model.GNNModel.KBAT.KBAT method) init_emb() (neuralkg.model.GNNModel.XTransE.XTransE method) (neuralkg.model.KGEModel.BoxE.BoxE method) (neuralkg.model.KGEModel.ComplEx.ComplEx method) (neuralkg.model.KGEModel.ConvE.ConvE method) (neuralkg.model.KGEModel.CrossE.CrossE method) (neuralkg.model.KGEModel.DistMult.DistMult method) (neuralkg.model.KGEModel.model.Model method) (neuralkg.model.KGEModel.RotatE.RotatE method) (neuralkg.model.KGEModel.SimplE.SimplE method) (neuralkg.model.KGEModel.TransE.TransE method) (neuralkg.model.KGEModel.TransH.TransH method) (neuralkg.model.KGEModel.TransR.TransR method) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg.model.RuleModel.IterE.IterE method) (neuralkg.model.RuleModel.RugE.RugE method)

init_GAT_emb() (neuralkg.model.GNNModel.KBAT.KBAT method) init_label() (neuralkg.data.Sampler.CrossESampler method) init_model() (neuralkg.model.GNNModel.CompGCN.CompGCN method) IterE (class in neuralkg.model.RuleModel.IterE) IterE_Loss (class in neuralkg.loss.IterE_Loss) IterELitModel (class in neuralkg.lit_model.IterELitModel)

K

KBAT (class in neuralkg.model.GNNModel.KBAT) KBAT_Loss (class in neuralkg.loss.KBAT_Loss) KBATLitModel (class in neuralkg.lit_model.KBATLitModel) KBATSampler (class in neuralkg.data.Sampler)

KGData (class in neuralkg.data.DataPreprocess) KGDataModule (class in neuralkg.data.KGDataModule) KGDataset (class in neuralkg.data.Sampler) KGELitModel (class in neuralkg.lit_model.KGELitModel)

L

l2_loss() (neuralkg.model.KGEModel.SimplE.SimplE method) label (neuralkg.data.Sampler.CompGCNSampler attribute) (neuralkg.data.Sampler.ConvSampler attribute) (neuralkg.data.Sampler.GraphSampler attribute) (neuralkg.data.Sampler.KBATSampler attribute)

lef_mean (neuralkg.data.Sampler.BernSampler attribute)

M

margin (neuralkg.model.KGEModel.DistMult.DistMult attribute) (neuralkg.model.KGEModel.RotatE.RotatE attribute) (neuralkg.model.KGEModel.SimplE.SimplE attribute) (neuralkg.model.KGEModel.TransE.TransE attribute) (neuralkg.model.KGEModel.TransH.TransH attribute) (neuralkg.model.KGEModel.TransR.TransR attribute) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg.model.RuleModel.IterE.IterE attribute) (neuralkg.model.RuleModel.RugE.RugE attribute) Margin_Loss (class in neuralkg.loss.Margin_Loss) max_neighbor (neuralkg.data.Sampler.XTransESampler attribute) message_func() (neuralkg.model.GNNModel.CompGCN.CompGCNCov method) Model (class in neuralkg.model.KGEModel.model) model (neuralkg.loss.Adv_Loss.Adv_Loss attribute) module neuralkg.data.base_data_module neuralkg.data.DataPreprocess neuralkg.data.Grounding neuralkg.data.KGDataModule neuralkg.data.RuleDataLoader neuralkg.data.Sampler neuralkg.eval_task.RGCN_prediction neuralkg.lit_model.BaseLitModel neuralkg.lit_model.CompGCNLitModel neuralkg.lit_model.ConvELitModel neuralkg.lit_model.CrossELitModel neuralkg.lit_model.IterELitModel neuralkg.lit_model.KBATLitModel neuralkg.lit_model.KGELitModel neuralkg.lit_model.RGCNLitModel neuralkg.lit_model.RugELitModel neuralkg.lit_model.XTransELitModel neuralkg.loss.Adv_Loss neuralkg.loss.ComplEx_NNE_AER_Loss neuralkg.loss.Cross_Entropy_Loss neuralkg.loss.CrossE_Loss neuralkg.loss.IterE_Loss neuralkg.loss.KBAT_Loss neuralkg.loss.Margin_Loss neuralkg.loss.RGCN_Loss neuralkg.loss.RugE_Loss neuralkg.loss.SimplE_Loss neuralkg.model.GNNModel.CompGCN neuralkg.model.GNNModel.KBAT neuralkg.model.GNNModel.RGCN neuralkg.model.GNNModel.XTransE neuralkg.model.KGEModel.BoxE neuralkg.model.KGEModel.ComplEx neuralkg.model.KGEModel.ConvE neuralkg.model.KGEModel.CrossE neuralkg.model.KGEModel.DistMult neuralkg.model.KGEModel.model neuralkg.model.KGEModel.RotatE neuralkg.model.KGEModel.SimplE neuralkg.model.KGEModel.TransE neuralkg.model.KGEModel.TransH neuralkg.model.KGEModel.TransR neuralkg.model.RuleModel.ComplEx_NNE_AER neuralkg.model.RuleModel.IterE neuralkg.model.RuleModel.RugE

N

n_hop (neuralkg.data.Sampler.KBATSampler attribute) neg_sample (neuralkg.data.Sampler.XTransESampler attribute) neighbours (neuralkg.data.Sampler.KBATSampler attribute) neuralkg.data.base_data_module module neuralkg.data.DataPreprocess module neuralkg.data.Grounding module neuralkg.data.KGDataModule module neuralkg.data.RuleDataLoader module neuralkg.data.Sampler module neuralkg.eval_task.RGCN_prediction module neuralkg.lit_model.BaseLitModel module neuralkg.lit_model.CompGCNLitModel module neuralkg.lit_model.ConvELitModel module neuralkg.lit_model.CrossELitModel module neuralkg.lit_model.IterELitModel module neuralkg.lit_model.KBATLitModel module neuralkg.lit_model.KGELitModel module neuralkg.lit_model.RGCNLitModel module neuralkg.lit_model.RugELitModel module neuralkg.lit_model.XTransELitModel module neuralkg.loss.Adv_Loss module neuralkg.loss.ComplEx_NNE_AER_Loss module neuralkg.loss.Cross_Entropy_Loss module neuralkg.loss.CrossE_Loss module neuralkg.loss.IterE_Loss module neuralkg.loss.KBAT_Loss module neuralkg.loss.Margin_Loss module neuralkg.loss.RGCN_Loss module

neuralkg.loss.RugE_Loss module neuralkg.loss.SimplE_Loss module neuralkg.model.GNNModel.CompGCN module neuralkg.model.GNNModel.KBAT module neuralkg.model.GNNModel.RGCN module neuralkg.model.GNNModel.XTransE module neuralkg.model.KGEModel.BoxE module neuralkg.model.KGEModel.ComplEx module neuralkg.model.KGEModel.ConvE module neuralkg.model.KGEModel.CrossE module neuralkg.model.KGEModel.DistMult module neuralkg.model.KGEModel.model module neuralkg.model.KGEModel.RotatE module neuralkg.model.KGEModel.SimplE module neuralkg.model.KGEModel.TransE module neuralkg.model.KGEModel.TransH module neuralkg.model.KGEModel.TransR module neuralkg.model.RuleModel.ComplEx_NNE_AER module neuralkg.model.RuleModel.IterE module neuralkg.model.RuleModel.RugE module node_norm_to_edge_norm() (neuralkg.data.Sampler.CompGCNSampler method) (neuralkg.data.Sampler.GraphSampler method) norm (neuralkg.data.Sampler.CompGCNSampler attribute) (neuralkg.data.Sampler.GraphSampler attribute) norm_vector (neuralkg.model.KGEModel.TransH.TransH attribute) normal() (in module neuralkg.data.Sampler) normalize() (neuralkg.loss.Adv_Loss.Adv_Loss method) (neuralkg.loss.IterE_Loss.IterE_Loss method) num_ent (neuralkg.data.Sampler.CompGCNTestSampler attribute) (neuralkg.data.Sampler.GraphTestSampler attribute) (neuralkg.data.Sampler.TestSampler attribute) num_workers (neuralkg.data.RuleDataLoader.RuleDataLoader attribute)

P

pin_memory (neuralkg.data.RuleDataLoader.RuleDataLoader attribute) precision (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg.lit_model.IterELitModel.IterELitModel attribute) (neuralkg.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg.lit_model.RugELitModel.RugELitModel attribute) (neuralkg.lit_model.XTransELitModel.XTransELitModel attribute) prefetch_factor (neuralkg.data.RuleDataLoader.RuleDataLoader attribute)

prepare_data() (neuralkg.data.base_data_module.BaseDataModule method) (neuralkg.data.KGDataModule.KGDataModule method) prepare_data_per_node (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg.lit_model.IterELitModel.IterELitModel attribute) (neuralkg.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg.lit_model.RugELitModel.RugELitModel attribute) (neuralkg.lit_model.XTransELitModel.XTransELitModel attribute) PropositionalizeRule() (neuralkg.data.Grounding.GroundAllRules method)

R

readData() (neuralkg.data.Grounding.GroundAllRules method) reduce_func() (neuralkg.model.GNNModel.CompGCN.CompGCNCov method) reg_loss() (neuralkg.loss.RGCN_Loss.RGCN_Loss method) regularize_loss() (neuralkg.model.KGEModel.CrossE.CrossE method) rel2id (neuralkg.data.DataPreprocess.KGData attribute) rel_emb (neuralkg.model.KGEModel.CrossE.CrossE attribute) (neuralkg.model.KGEModel.DistMult.DistMult attribute) (neuralkg.model.KGEModel.RotatE.RotatE attribute) (neuralkg.model.KGEModel.SimplE.SimplE attribute) (neuralkg.model.KGEModel.TransE.TransE attribute) (neuralkg.model.KGEModel.TransH.TransH attribute) (neuralkg.model.KGEModel.TransR.TransR attribute) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg.model.RuleModel.IterE.IterE attribute) (neuralkg.model.RuleModel.RugE.RugE attribute) rel_inv_emb (neuralkg.model.KGEModel.SimplE.SimplE attribute) rel_reverse_emb (neuralkg.model.KGEModel.CrossE.CrossE attribute) relation (neuralkg.data.Sampler.CompGCNSampler attribute) (neuralkg.data.Sampler.GraphSampler attribute)

Relation2Tuple (neuralkg.data.DataPreprocess.KGData attribute) RelSub2Obj (neuralkg.data.DataPreprocess.KGData attribute) RevSampler (class in neuralkg.data.DataPreprocess) RGCN (class in neuralkg.model.GNNModel.RGCN) RGCN_Loss (class in neuralkg.loss.RGCN_Loss) RGCN_predict() (in module neuralkg.eval_task.RGCN_prediction) RGCNLitModel (class in neuralkg.lit_model.RGCNLitModel) rig_mean (neuralkg.data.Sampler.BernSampler attribute) RotatE (class in neuralkg.model.KGEModel.RotatE) rt2h_all (neuralkg.data.Sampler.CompGCNTestSampler attribute) (neuralkg.data.Sampler.GraphTestSampler attribute) (neuralkg.data.Sampler.TestSampler attribute) rt2h_train (neuralkg.data.DataPreprocess.KGData attribute) (neuralkg.data.DataPreprocess.RevSampler attribute) RugE (class in neuralkg.model.RuleModel.RugE) RugE_Loss (class in neuralkg.loss.RugE_Loss) RugELitModel (class in neuralkg.lit_model.RugELitModel) RuleDataLoader (class in neuralkg.data.RuleDataLoader) RuleDataset (class in neuralkg.data.RuleDataLoader) run_axiom_probability() (neuralkg.model.RuleModel.IterE.IterE method)

S

sam_negative() (neuralkg.data.Sampler.KBATSampler method) sampler (neuralkg.data.RuleDataLoader.RuleDataLoader attribute) (neuralkg.data.Sampler.CompGCNTestSampler attribute) (neuralkg.data.Sampler.GraphTestSampler attribute) (neuralkg.data.Sampler.TestSampler attribute) sampling() (neuralkg.data.Sampler.AdvSampler method) (neuralkg.data.Sampler.AllSampler method) (neuralkg.data.Sampler.BernSampler method) (neuralkg.data.Sampler.CompGCNSampler method) (neuralkg.data.Sampler.CompGCNTestSampler method) (neuralkg.data.Sampler.ConvSampler method) (neuralkg.data.Sampler.CrossESampler method) (neuralkg.data.Sampler.GraphSampler method) (neuralkg.data.Sampler.GraphTestSampler method) (neuralkg.data.Sampler.KBATSampler method) (neuralkg.data.Sampler.TestSampler method) (neuralkg.data.Sampler.UniSampler method) (neuralkg.data.Sampler.XTransESampler method) sampling_keys() (neuralkg.data.Sampler.AllSampler method) (neuralkg.data.Sampler.BernSampler static method) (neuralkg.data.Sampler.ConvSampler method) (neuralkg.data.Sampler.CrossESampler method) sampling_negative() (neuralkg.data.Sampler.GraphSampler method) (neuralkg.data.Sampler.KBATSampler method)

sampling_positive() (neuralkg.data.Sampler.GraphSampler method) score_func() (neuralkg.model.GNNModel.XTransE.XTransE method) (neuralkg.model.KGEModel.BoxE.BoxE method) (neuralkg.model.KGEModel.ComplEx.ComplEx method) (neuralkg.model.KGEModel.ConvE.ConvE method) (neuralkg.model.KGEModel.CrossE.CrossE method) (neuralkg.model.KGEModel.DistMult.DistMult method) (neuralkg.model.KGEModel.model.Model method) (neuralkg.model.KGEModel.RotatE.RotatE method) (neuralkg.model.KGEModel.SimplE.SimplE method) (neuralkg.model.KGEModel.TransE.TransE method) (neuralkg.model.KGEModel.TransH.TransH method) (neuralkg.model.KGEModel.TransR.TransR method) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER method) (neuralkg.model.RuleModel.IterE.IterE method) (neuralkg.model.RuleModel.RugE.RugE method) setup() (neuralkg.data.base_data_module.BaseDataModule method) (neuralkg.data.KGDataModule.KGDataModule method) sim() (neuralkg.model.RuleModel.IterE.IterE method) SimplE (class in neuralkg.model.KGEModel.SimplE) SimplE_Loss (class in neuralkg.loss.SimplE_Loss) SpecialSpmmFinal (class in neuralkg.model.GNNModel.KBAT) SpecialSpmmFunctionFinal (class in neuralkg.model.GNNModel.KBAT) split_embedding() (neuralkg.model.RuleModel.IterE.IterE method)

T

t2rh_train (neuralkg.data.DataPreprocess.KGData attribute) t_weighted_vector (neuralkg.model.KGEModel.CrossE.CrossE attribute) tail_batch() (neuralkg.data.DataPreprocess.BaseSampler method) (neuralkg.data.DataPreprocess.RevSampler method) tail_predict() (in module neuralkg.eval_task.RGCN_prediction) test_dataloader() (neuralkg.data.base_data_module.BaseDataModule method) (neuralkg.data.KGDataModule.KGDataModule method) test_epoch_end() (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg.lit_model.ConvELitModel.ConvELitModel method) (neuralkg.lit_model.CrossELitModel.CrossELitModel method) (neuralkg.lit_model.IterELitModel.IterELitModel method) (neuralkg.lit_model.KBATLitModel.KBATLitModel method) (neuralkg.lit_model.KGELitModel.KGELitModel method) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg.lit_model.RugELitModel.RugELitModel method) (neuralkg.lit_model.XTransELitModel.XTransELitModel method) test_step() (neuralkg.lit_model.BaseLitModel.BaseLitModel method) (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg.lit_model.ConvELitModel.ConvELitModel method) (neuralkg.lit_model.CrossELitModel.CrossELitModel method) (neuralkg.lit_model.IterELitModel.IterELitModel method) (neuralkg.lit_model.KBATLitModel.KBATLitModel method) (neuralkg.lit_model.KGELitModel.KGELitModel method) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg.lit_model.RugELitModel.RugELitModel method) (neuralkg.lit_model.XTransELitModel.XTransELitModel method) test_triples (neuralkg.data.DataPreprocess.KGData attribute) TestSampler (class in neuralkg.data.Sampler) timeout (neuralkg.data.RuleDataLoader.RuleDataLoader attribute) tr_bias (neuralkg.model.KGEModel.CrossE.CrossE attribute) train_dataloader() (neuralkg.data.base_data_module.BaseDataModule method) (neuralkg.data.KGDataModule.KGDataModule method) train_triples (neuralkg.data.DataPreprocess.KGData attribute) training (neuralkg.lit_model.BaseLitModel.BaseLitModel attribute) (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg.lit_model.IterELitModel.IterELitModel attribute) (neuralkg.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg.lit_model.KGELitModel.KGELitModel attribute) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg.lit_model.RugELitModel.RugELitModel attribute) (neuralkg.lit_model.XTransELitModel.XTransELitModel attribute) (neuralkg.loss.Adv_Loss.Adv_Loss attribute) (neuralkg.loss.ComplEx_NNE_AER_Loss.ComplEx_NNE_AER_Loss attribute) (neuralkg.loss.Cross_Entropy_Loss.Cross_Entropy_Loss attribute) (neuralkg.loss.CrossE_Loss.CrossE_Loss attribute) (neuralkg.loss.IterE_Loss.IterE_Loss attribute) (neuralkg.loss.KBAT_Loss.KBAT_Loss attribute) (neuralkg.loss.Margin_Loss.Margin_Loss attribute) (neuralkg.loss.RGCN_Loss.RGCN_Loss attribute) (neuralkg.loss.RugE_Loss.RugE_Loss attribute) (neuralkg.loss.SimplE_Loss.SimplE_Loss attribute) (neuralkg.model.GNNModel.CompGCN.CompGCN attribute) (neuralkg.model.GNNModel.CompGCN.CompGCNCov attribute) (neuralkg.model.GNNModel.KBAT.GraphAttentionLayer attribute) (neuralkg.model.GNNModel.KBAT.KBAT attribute) (neuralkg.model.GNNModel.KBAT.SpecialSpmmFinal attribute) (neuralkg.model.GNNModel.RGCN.RGCN attribute) (neuralkg.model.GNNModel.XTransE.XTransE attribute) (neuralkg.model.KGEModel.BoxE.BoxE attribute) (neuralkg.model.KGEModel.ComplEx.ComplEx attribute) (neuralkg.model.KGEModel.ConvE.ConvE attribute) (neuralkg.model.KGEModel.CrossE.CrossE attribute) (neuralkg.model.KGEModel.DistMult.DistMult attribute) (neuralkg.model.KGEModel.model.Model attribute) (neuralkg.model.KGEModel.RotatE.RotatE attribute) (neuralkg.model.KGEModel.SimplE.SimplE attribute) (neuralkg.model.KGEModel.TransE.TransE attribute) (neuralkg.model.KGEModel.TransH.TransH attribute) (neuralkg.model.KGEModel.TransR.TransR attribute) (neuralkg.model.RuleModel.ComplEx_NNE_AER.ComplEx_NNE_AER attribute) (neuralkg.model.RuleModel.IterE.IterE attribute) (neuralkg.model.RuleModel.RugE.RugE attribute)

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

U

uni_sampling() (neuralkg.data.Sampler.UniSampler method) UniSampler (class in neuralkg.data.Sampler) update_axiom() (neuralkg.lit_model.IterELitModel.IterELitModel method) update_train_triples() (neuralkg.model.RuleModel.IterE.IterE method) update_valid_axioms() (neuralkg.model.RuleModel.IterE.IterE method) use_amp (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel attribute) (neuralkg.lit_model.ConvELitModel.ConvELitModel attribute) (neuralkg.lit_model.CrossELitModel.CrossELitModel attribute) (neuralkg.lit_model.IterELitModel.IterELitModel attribute) (neuralkg.lit_model.KBATLitModel.KBATLitModel attribute) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel attribute) (neuralkg.lit_model.RugELitModel.RugELitModel attribute) (neuralkg.lit_model.XTransELitModel.XTransELitModel attribute)

V

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

validation_step() (neuralkg.lit_model.BaseLitModel.BaseLitModel method) (neuralkg.lit_model.CompGCNLitModel.CompGCNLitModel method) (neuralkg.lit_model.ConvELitModel.ConvELitModel method) (neuralkg.lit_model.CrossELitModel.CrossELitModel method) (neuralkg.lit_model.IterELitModel.IterELitModel method) (neuralkg.lit_model.KBATLitModel.KBATLitModel method) (neuralkg.lit_model.KGELitModel.KGELitModel method) (neuralkg.lit_model.RGCNLitModel.RGCNLitModel method) (neuralkg.lit_model.RugELitModel.RugELitModel method) (neuralkg.lit_model.XTransELitModel.XTransELitModel method)

X

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

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