redkg package

Subpackages

• redkg.models package
  • Submodules
    • MLP
      • MLP.forward()
    • Net
      • Net.forward()
    • GAT
      • GAT.forward()
      • GAT.predict_edges()
    • GCN
      • GCN.forward()
      • GCN.predict_edges()
    • AbstractLayer
      • AbstractLayer.distance()
      • AbstractLayer.forward_gcn()
      • AbstractLayer.get_n_hop()
    • GCNGRU
      • GCNGRU.forward()
    • GRU
      • GRU.forward()
      • GRU.transe_forward()
    • GraphConvolution
      • GraphConvolution.forward()
      • GraphConvolution.reset_parameters()
      • GraphConvolution.weight
    • GraphSAGE
      • GraphSAGE.forward()
      • GraphSAGE.predict_edges()
    • KGEModel
      • KGEModel.ComplEx()
      • KGEModel.DistMult()
      • KGEModel.RotatE()
      • KGEModel.TransE()
      • KGEModel.forward()
      • KGEModel.test_step()
      • KGEModel.train_step()

Submodules

• Config
• BidirectionalOneShotIterator
  • BidirectionalOneShotIterator.one_shot_iterator()
• TestDataset
  • TestDataset.collate_fn()
• TrainDataset
  • TrainDataset.collate_fn()
• get_info()
• Simulator
  • Simulator.get_user_data()
  • Simulator.step()
• Evaluator
  • Evaluator.eval()
• generate_subgraphs()
• generate_subgraphs_dataset()
• common_neighbors()
• generate_negative_samples()
• DataPreprocessor
  • DataPreprocessor.process_data()
• TrainPipeline
  • TrainPipeline.memory_sampling()
  • TrainPipeline.optimize_model()
  • TrainPipeline.run()
  • TrainPipeline.tmp_Q_eps_greedy()
• train_gnn_model()
• train_kge_model()
• AttributeDict
• pickle_dump()
• pickle_load()