NeuralKG-ind

• Data
  • DataPreprocess
    • KGData
      • KGData.args
      • KGData.ent2id
      • KGData.rel2id
      • KGData.id2ent
      • KGData.id2rel
      • KGData.train_triples
      • KGData.valid_triples
      • KGData.test_triples
      • KGData.all_true_triples
      • KGData.TrainTriples
      • KGData.Relation2Tuple
      • KGData.RelSub2Obj
      • KGData.hr2t_train
      • KGData.rt2h_train
      • KGData.h2rt_train
      • KGData.t2rh_train
      • KGData.get_id()
      • KGData.get_triples_id()
      • KGData.get_hr2t_rt2h_from_train()
      • KGData.count_frequency()
      • KGData.get_h2rt_t2hr_from_train()
      • KGData.get_hr_trian()
    • GRData
      • GRData.args
      • GRData.db_name_pos
      • GRData.db_name_neg
      • GRData.m_h2r
      • GRData.m_t2r
      • GRData.ssp_graph
      • GRData.graph
      • GRData.id2entity
      • GRData.id2relation
      • GRData.load_data_grail()
      • GRData.load_ind_data_grail()
      • GRData.generate_train()
      • GRData.generate_valid()
      • GRData.prepare_subgraphs()
      • GRData.prepare_features_new()
    • MetaTrainGRData
      • MetaTrainGRData.subgraphs_db
    • MetaValidGRData
      • MetaValidGRData.subgraphs_db
    • KGEEvalData
      • KGEEvalData.triples
      • KGEEvalData.num_ent
      • KGEEvalData.hr2t
      • KGEEvalData.rt2h
      • KGEEvalData.num_cand
      • KGEEvalData.get_label()
      • KGEEvalData.collate_fn()
    • BaseSampler
      • BaseSampler.corrupt_head()
      • BaseSampler.corrupt_tail()
      • BaseSampler.head_batch()
      • BaseSampler.tail_batch()
      • BaseSampler.get_train()
      • BaseSampler.get_valid()
      • BaseSampler.get_test()
      • BaseSampler.get_all_true_triples()
    • RevSampler
      • RevSampler.hr2t_train
      • RevSampler.rt2h_train
      • RevSampler.add_reverse_relation()
      • RevSampler.add_reverse_triples()
      • RevSampler.get_train()
      • RevSampler.get_valid()
      • RevSampler.get_test()
      • RevSampler.get_all_true_triples()
      • RevSampler.corrupt_head()
      • RevSampler.corrupt_tail()
      • RevSampler.head_batch()
      • RevSampler.tail_batch()
    • BaseGraph
      • BaseGraph.get_train()
      • BaseGraph.get_valid()
      • BaseGraph.get_test()
      • BaseGraph.generate_ind_test()
      • BaseGraph.load_data_grail_ind()
      • BaseGraph.get_neg_samples_replacing_head_tail()
    • BaseMeta
      • BaseMeta.get_train()
      • BaseMeta.get_valid()
      • BaseMeta.get_test()
  • Grounding
    • GroundAllRules
      • GroundAllRules.PropositionalizeRule()
      • GroundAllRules.readData()
      • GroundAllRules.groundRule()
  • KGDataModule
    • KGDataModule
      • KGDataModule.get_data_config()
      • KGDataModule.prepare_data()
      • KGDataModule.setup()
      • KGDataModule.get_train_bs()
      • KGDataModule.train_dataloader()
      • KGDataModule.val_dataloader()
      • KGDataModule.test_dataloader()
  • RuleDataLoader
    • RuleDataset
    • RuleDataLoader
      • RuleDataLoader.dataset
      • RuleDataLoader.batch_size
      • RuleDataLoader.num_workers
      • RuleDataLoader.pin_memory
      • RuleDataLoader.drop_last
      • RuleDataLoader.timeout
      • RuleDataLoader.sampler
      • RuleDataLoader.prefetch_factor
  • Sampler
    • SubSampler
      • SubSampler.sampling()
      • SubSampler.get_sampling_keys()
    • RMPISampler
      • RMPISampler.sampling()
      • RMPISampler.get_sampling_keys()
    • UniSampler
      • UniSampler.cross_sampling_flag
      • UniSampler.sampling()
      • UniSampler.uni_sampling()
      • UniSampler.get_sampling_keys()
    • BernSampler
      • BernSampler.lef_mean
      • BernSampler.rig_mean
      • BernSampler.sampling()
      • BernSampler.calc_bern()
      • BernSampler.sampling_keys()
    • AdvSampler
      • AdvSampler.sampling()
      • AdvSampler.calc_freq()
    • AllSampler
      • AllSampler.sampling()
      • AllSampler.sampling_keys()
    • CrossESampler
      • CrossESampler.sampling()
      • CrossESampler.init_label()
      • CrossESampler.sampling_keys()
    • ConvSampler
      • ConvSampler.label
      • ConvSampler.triples
      • ConvSampler.sampling()
      • ConvSampler.sampling_keys()
    • XTransESampler
      • XTransESampler.triples
      • XTransESampler.neg_sample
      • XTransESampler.h_neighbor
      • XTransESampler.h_mask
      • XTransESampler.max_neighbor
      • XTransESampler.sampling()
      • XTransESampler.get_sampling_keys()
    • GraphSampler
      • GraphSampler.entity
      • GraphSampler.relation
      • GraphSampler.triples
      • GraphSampler.graph
      • GraphSampler.norm
      • GraphSampler.label
      • GraphSampler.sampling()
      • GraphSampler.get_sampling_keys()
      • GraphSampler.sampling_negative()
      • GraphSampler.build_graph()
      • GraphSampler.comp_deg_norm()
      • GraphSampler.node_norm_to_edge_norm()
      • GraphSampler.sampling_positive()
    • KBATSampler
      • KBATSampler.n_hop
      • KBATSampler.graph
      • KBATSampler.neighbours
      • KBATSampler.adj_matrix
      • KBATSampler.triples
      • KBATSampler.triples_GAT_pos
      • KBATSampler.triples_GAT_neg
      • KBATSampler.triples_Con
      • KBATSampler.label
      • KBATSampler.sampling()
      • KBATSampler.get_sampling_keys()
      • KBATSampler.bfs()
      • KBATSampler.get_neighbors()
      • KBATSampler.get_unique_entity()
      • KBATSampler.get_batch_nhop_neighbors_all()
      • KBATSampler.sampling_negative()
      • KBATSampler.sam_negative()
    • CompGCNSampler
      • CompGCNSampler.relation
      • CompGCNSampler.triples
      • CompGCNSampler.graph
      • CompGCNSampler.norm
      • CompGCNSampler.label
      • CompGCNSampler.sampling()
      • CompGCNSampler.get_sampling_keys()
      • CompGCNSampler.node_norm_to_edge_norm()
    • TestSampler
      • TestSampler.sampler
      • TestSampler.hr2t_all
      • TestSampler.rt2h_all
      • TestSampler.num_ent
      • TestSampler.get_hr2t_rt2h_from_all()
      • TestSampler.sampling()
      • TestSampler.get_sampling_keys()
    • ValidSampler
      • ValidSampler.sampler
      • ValidSampler.args
      • ValidSampler.sampling()
      • ValidSampler.get_sampling_keys()
    • ValidRMPISampler
      • ValidRMPISampler.sampler
      • ValidRMPISampler.args
      • ValidRMPISampler.sampling()
      • ValidRMPISampler.get_sampling_keys()
    • TestSampler_hit
      • TestSampler_hit.sampler
      • TestSampler_hit.args
      • TestSampler_hit.m_h2r
      • TestSampler_hit.m_t2r
      • TestSampler_hit.sampling()
      • TestSampler_hit.get_sampling_keys()
      • TestSampler_hit.get_subgraphs()
      • TestSampler_hit.prepare_features()
    • TestRMPISampler_hit
      • TestRMPISampler_hit.sampler
      • TestRMPISampler_hit.args
      • TestRMPISampler_hit.sampling()
      • TestRMPISampler_hit.get_sampling_keys()
      • TestRMPISampler_hit.prepare_subgraph()
      • TestRMPISampler_hit.get_subgraphs()
      • TestRMPISampler_hit.prepare_features()
    • TestSampler_auc
      • TestSampler_auc.sampler
      • TestSampler_auc.args
      • TestSampler_auc.sampling()
    • TestRMPISampler_auc
      • TestRMPISampler_auc.sampler
      • TestRMPISampler_auc.args
      • TestRMPISampler_auc.sampling()
      • TestRMPISampler_auc.get_sampling_keys()
    • MetaSampler
      • MetaSampler.sampling()
      • MetaSampler.get_sampling_keys()
    • ValidMetaSampler
      • ValidMetaSampler.sampling()
      • ValidMetaSampler.get_sampling_keys()
    • TestMetaSampler_hit
      • TestMetaSampler_hit.sampling()
      • TestMetaSampler_hit.get_sampling_keys()
    • TestMetaSampler_auc
      • TestMetaSampler_auc.sampling()
      • TestMetaSampler_auc.get_sampling_keys()
    • GraphTestSampler
      • GraphTestSampler.sampler
      • GraphTestSampler.hr2t_all
      • GraphTestSampler.rt2h_all
      • GraphTestSampler.num_ent
      • GraphTestSampler.triples
      • GraphTestSampler.get_hr2t_rt2h_from_all()
      • GraphTestSampler.sampling()
      • GraphTestSampler.get_sampling_keys()
    • CompGCNTestSampler
      • CompGCNTestSampler.sampler
      • CompGCNTestSampler.hr2t_all
      • CompGCNTestSampler.rt2h_all
      • CompGCNTestSampler.num_ent
      • CompGCNTestSampler.triples
      • CompGCNTestSampler.get_hr2t_rt2h_from_all()
      • CompGCNTestSampler.sampling()
      • CompGCNTestSampler.get_sampling_keys()
    • SEGNNTrainProcess
      • SEGNNTrainProcess.get_h2rt_t2hr_from_train()
      • SEGNNTrainProcess.get_onehot_label()
      • SEGNNTrainProcess.get_sampling()
      • SEGNNTrainProcess.construct_kg()
    • SEGNNTrainSampler
      • SEGNNTrainSampler.get_train()
      • SEGNNTrainSampler.get_valid()
      • SEGNNTrainSampler.get_test()
      • SEGNNTrainSampler.sampling()
    • SEGNNTestSampler
      • SEGNNTestSampler.get_hr2t_rt2h_from_all()
      • SEGNNTestSampler.sampling()
      • SEGNNTestSampler.get_sampling_keys()
    • KGDataset
  • base_data_module
    • Config
    • BaseDataModule
      • BaseDataModule.add_to_argparse()
      • BaseDataModule.prepare_data()
      • BaseDataModule.setup()
      • BaseDataModule.train_dataloader()
      • BaseDataModule.val_dataloader()
      • BaseDataModule.test_dataloader()
      • BaseDataModule.get_config()
• Eval_Task
  • link_prediction
    • link_predict()
    • ind_predict()
    • head_predict()
    • tail_predict()
    • calc_ranks()
    • link_predict_SEGNN()
    • head_predict_SEGNN()
    • tail_predict_SEGNN()
    • calc_ranks_SEGNN()
  • triple_classification
    • classification()
• Lit_Model
  • BaseLitModel
    • Config
    • BaseLitModel
      • BaseLitModel.add_to_argparse()
      • BaseLitModel.configure_optimizers()
      • BaseLitModel.forward()
      • BaseLitModel.training_step()
      • BaseLitModel.validation_step()
      • BaseLitModel.test_step()
      • BaseLitModel.get_results()
      • BaseLitModel.get_auc()
      • BaseLitModel.training
  • KGELitModel
    • KGELitModel
      • KGELitModel.forward()
      • KGELitModel.add_to_argparse()
      • KGELitModel.training_step()
      • KGELitModel.validation_step()
      • KGELitModel.validation_epoch_end()
      • KGELitModel.test_step()
      • KGELitModel.test_epoch_end()
      • KGELitModel.configure_optimizers()
      • KGELitModel.training
      • KGELitModel.use_amp
      • KGELitModel.precision
      • KGELitModel.prepare_data_per_node
      • KGELitModel.allow_zero_length_dataloader_with_multiple_devices
  • indGNNLitModel
    • indGNNLitModel
      • indGNNLitModel.forward()
      • indGNNLitModel.add_to_argparse()
      • indGNNLitModel.training_step()
      • indGNNLitModel.validation_step()
      • indGNNLitModel.validation_epoch_end()
      • indGNNLitModel.test_step()
      • indGNNLitModel.test_epoch_end()
      • indGNNLitModel.configure_optimizers()
      • indGNNLitModel.training
      • indGNNLitModel.use_amp
      • indGNNLitModel.precision
      • indGNNLitModel.prepare_data_per_node
      • indGNNLitModel.allow_zero_length_dataloader_with_multiple_devices
  • CompGCNLitModel
    • CompGCNLitModel
      • CompGCNLitModel.forward()
      • CompGCNLitModel.training_step()
      • CompGCNLitModel.validation_step()
      • CompGCNLitModel.validation_epoch_end()
      • CompGCNLitModel.test_step()
      • CompGCNLitModel.test_epoch_end()
      • CompGCNLitModel.configure_optimizers()
      • CompGCNLitModel.training
      • CompGCNLitModel.use_amp
      • CompGCNLitModel.precision
      • CompGCNLitModel.prepare_data_per_node
      • CompGCNLitModel.allow_zero_length_dataloader_with_multiple_devices
  • ConvELitModel
    • ConvELitModel
      • ConvELitModel.forward()
      • ConvELitModel.training_step()
      • ConvELitModel.validation_step()
      • ConvELitModel.validation_epoch_end()
      • ConvELitModel.test_step()
      • ConvELitModel.test_epoch_end()
      • ConvELitModel.configure_optimizers()
      • ConvELitModel.training
      • ConvELitModel.use_amp
      • ConvELitModel.precision
      • ConvELitModel.prepare_data_per_node
      • ConvELitModel.allow_zero_length_dataloader_with_multiple_devices
  • CrossELitModel
    • CrossELitModel
      • CrossELitModel.forward()
      • CrossELitModel.training_step()
      • CrossELitModel.validation_step()
      • CrossELitModel.validation_epoch_end()
      • CrossELitModel.test_step()
      • CrossELitModel.test_epoch_end()
      • CrossELitModel.configure_optimizers()
      • CrossELitModel.training
      • CrossELitModel.use_amp
      • CrossELitModel.precision
      • CrossELitModel.prepare_data_per_node
      • CrossELitModel.allow_zero_length_dataloader_with_multiple_devices
  • IterELitModel
    • IterELitModel
      • IterELitModel.forward()
      • IterELitModel.training_step()
      • IterELitModel.training_epoch_end()
      • IterELitModel.update_axiom()
      • IterELitModel.validation_step()
      • IterELitModel.validation_epoch_end()
      • IterELitModel.test_step()
      • IterELitModel.test_epoch_end()
      • IterELitModel.configure_optimizers()
      • IterELitModel.training
      • IterELitModel.use_amp
      • IterELitModel.precision
      • IterELitModel.prepare_data_per_node
      • IterELitModel.allow_zero_length_dataloader_with_multiple_devices
  • KBATLitModel
    • KBATLitModel
      • KBATLitModel.forward()
      • KBATLitModel.training_step()
      • KBATLitModel.validation_step()
      • KBATLitModel.validation_epoch_end()
      • KBATLitModel.test_step()
      • KBATLitModel.test_epoch_end()
      • KBATLitModel.configure_optimizers()
      • KBATLitModel.training
      • KBATLitModel.use_amp
      • KBATLitModel.precision
      • KBATLitModel.prepare_data_per_node
      • KBATLitModel.allow_zero_length_dataloader_with_multiple_devices
  • MetaGNNLitModel
    • MetaGNNLitModel
      • MetaGNNLitModel.forward()
      • MetaGNNLitModel.add_to_argparse()
      • MetaGNNLitModel.training_step()
      • MetaGNNLitModel.validation_step()
      • MetaGNNLitModel.validation_epoch_end()
      • MetaGNNLitModel.test_step()
      • MetaGNNLitModel.test_epoch_end()
      • MetaGNNLitModel.configure_optimizers()
      • MetaGNNLitModel.training
      • MetaGNNLitModel.use_amp
      • MetaGNNLitModel.precision
      • MetaGNNLitModel.prepare_data_per_node
      • MetaGNNLitModel.allow_zero_length_dataloader_with_multiple_devices
  • RGCNLitModel
    • RGCNLitModel
      • RGCNLitModel.forward()
      • RGCNLitModel.training_step()
      • RGCNLitModel.validation_step()
      • RGCNLitModel.validation_epoch_end()
      • RGCNLitModel.test_step()
      • RGCNLitModel.test_epoch_end()
      • RGCNLitModel.configure_optimizers()
      • RGCNLitModel.training
      • RGCNLitModel.use_amp
      • RGCNLitModel.precision
      • RGCNLitModel.prepare_data_per_node
      • RGCNLitModel.allow_zero_length_dataloader_with_multiple_devices
  • RugELitModel
    • RugELitModel
      • RugELitModel.forward()
      • RugELitModel.training_step()
      • RugELitModel.training_epoch_end()
      • RugELitModel.validation_step()
      • RugELitModel.validation_epoch_end()
      • RugELitModel.test_step()
      • RugELitModel.test_epoch_end()
      • RugELitModel.configure_optimizers()
      • RugELitModel.training
      • RugELitModel.use_amp
      • RugELitModel.precision
      • RugELitModel.prepare_data_per_node
      • RugELitModel.allow_zero_length_dataloader_with_multiple_devices
  • SEGNNLitModel
    • SEGNNLitModel
      • SEGNNLitModel.forward()
      • SEGNNLitModel.training_step()
      • SEGNNLitModel.validation_step()
      • SEGNNLitModel.validation_epoch_end()
      • SEGNNLitModel.test_step()
      • SEGNNLitModel.test_epoch_end()
      • SEGNNLitModel.get_kg()
      • SEGNNLitModel.configure_optimizers()
      • SEGNNLitModel.training
      • SEGNNLitModel.use_amp
      • SEGNNLitModel.precision
      • SEGNNLitModel.prepare_data_per_node
      • SEGNNLitModel.allow_zero_length_dataloader_with_multiple_devices
  • SNRILitModel
    • SNRILitModel
      • SNRILitModel.forward()
      • SNRILitModel.add_to_argparse()
      • SNRILitModel.training_step()
      • SNRILitModel.validation_step()
      • SNRILitModel.validation_epoch_end()
      • SNRILitModel.test_step()
      • SNRILitModel.test_epoch_end()
      • SNRILitModel.configure_optimizers()
      • SNRILitModel.training
      • SNRILitModel.use_amp
      • SNRILitModel.precision
      • SNRILitModel.prepare_data_per_node
      • SNRILitModel.allow_zero_length_dataloader_with_multiple_devices
  • XTransELitModel
    • XTransELitModel
      • XTransELitModel.forward()
      • XTransELitModel.training_step()
      • XTransELitModel.validation_step()
      • XTransELitModel.validation_epoch_end()
      • XTransELitModel.test_step()
      • XTransELitModel.test_epoch_end()
      • XTransELitModel.configure_optimizers()
      • XTransELitModel.training
      • XTransELitModel.use_amp
      • XTransELitModel.precision
      • XTransELitModel.prepare_data_per_node
      • XTransELitModel.allow_zero_length_dataloader_with_multiple_devices
• loss
  • Adv_Loss
    • Adv_Loss
      • Adv_Loss.args
      • Adv_Loss.model
      • Adv_Loss.forward()
      • Adv_Loss.normalize()
      • Adv_Loss.training
  • ComplEx_NNE_AER_Loss
    • ComplEx_NNE_AER_Loss
      • ComplEx_NNE_AER_Loss.forward()
      • ComplEx_NNE_AER_Loss.training
  • CrossE_Loss
    • CrossE_Loss
      • CrossE_Loss.forward()
      • CrossE_Loss.training
  • Cross_Entropy_Loss
    • Cross_Entropy_Loss
      • Cross_Entropy_Loss.forward()
      • Cross_Entropy_Loss.training
  • KBAT_Loss
    • KBAT_Loss
      • KBAT_Loss.forward()
      • KBAT_Loss.training
  • Logsig_Loss
    • Logsig_Loss
      • Logsig_Loss.forward()
      • Logsig_Loss.training
  • Margin_Loss
    • Margin_Loss
      • Margin_Loss.forward()
      • Margin_Loss.training
  • RGCN_Loss
    • RGCN_Loss
      • RGCN_Loss.reg_loss()
      • RGCN_Loss.forward()
      • RGCN_Loss.training
  • RugE_Loss
    • RugE_Loss
      • RugE_Loss.forward()
      • RugE_Loss.training
  • SimplE_Loss
    • SimplE_Loss
      • SimplE_Loss.forward()
      • SimplE_Loss.training
  • Softplus_Loss
    • Softplus_Loss
      • Softplus_Loss.model
      • Softplus_Loss.forward()
      • Softplus_Loss.training
• Model
  • KGEModel
    • model
      • Model
    • TransE
      • TransE
    • TransH
      • TransH
    • TransR
      • TransR
    • RotatE
      • RotatE
    • DistMult
      • DistMult
    • ComplEx
      • ComplEx
    • SimplE
      • SimplE
    • ConvE
      • ConvE
    • BoxE
      • BoxE
    • CrossE
      • CrossE
    • HAKE
      • HAKE
    • PairRE
      • PairRE
    • DualE
      • DualE
  • RuleModel
    • model
      • Model
    • ComplEx_NNE_AER
      • ComplEx_NNE_AER
    • IterE
      • IterE
    • RugE
      • RugE
  • GNNModel
    • model
      • Model
    • layer
      • GNNLayer
      • Aggregator
      • SUMAggregator
      • MLPAggregator
      • GRUAggregator
      • BatchGRU
    • RGCN
      • RGCN
      • RelGraphConv
    • CompGCN
      • CompGCN
      • CompGCNCov
    • SEGNN
      • SEGNN
      • CompLayer
      • NodeLayer
      • EdgeLayer
    • XTransE
      • XTransE
    • KBAT
      • KBAT
      • SpecialSpmmFunctionFinal
      • SpecialSpmmFinal
      • GraphAttentionLayer
    • Grail
      • Grail
      • RGCN
      • RelAttGraphConv
    • CoMPILE
      • CoMPILE
      • MySpMM
      • gnn_spmm()
    • SNRI
      • SNRI
      • RelCompGraphConv
      • Discriminator
    • RMPI
      • RMPI
    • MorsE
      • MorsE
      • EntInit
      • RelMorsGraphConv
      • RGCN
      • KGEModel
• Utils
  • setup_parser
    • setup_parser()
  • tools
    • import_class()
    • save_config()
    • load_config()
    • get_param()
    • deserialize()
    • deserialize_RMPI()
    • set_logger()
    • log_metrics()
    • log_step_metrics()
    • override_config()
    • reidx_withr_ande()
    • reidx()
    • reidx_withr()
    • data2pkl()
    • gen_subgraph_datasets()
    • load_ind_data_grail()
    • load_data_grail()
    • get_average_subgraph_size()
    • serialize()
    • sample_neg()
    • get_edge_count()
    • intialize_worker()
    • extract_save_subgraph()
    • links2subgraphs()
    • subgraph_extraction_labeling()
    • node_label()
    • remove_nodes()
    • get_neighbor_nodes()
    • incidence_matrix()
    • bfs_relational()
    • get_neighbors()
    • sp_row_vec_from_idx_list()
    • ssp_multigraph_to_dgl()
    • gen_meta_subgraph_datasets()
    • sample_one_subgraph()
    • get_average_meta_subgraph_size()
    • get_g()
    • get_g_bidir()
    • get_hr2t_rt2h()
    • get_hr2t_rt2h_sup_que()
    • get_indtest_test_dataset_and_train_g()