ReDKG

Contents:

  • redkg package
    • redkg.models package
      • MLP
      • Net
      • GAT
      • GCN
      • AbstractLayer
      • GCNGRU
      • GRU
      • GraphConvolution
      • GraphSAGE
      • KGEModel
    • Config
    • BidirectionalOneShotIterator
    • TestDataset
    • TrainDataset
    • get_info()
    • Simulator
    • Evaluator
    • generate_subgraphs()
    • generate_subgraphs_dataset()
    • common_neighbors()
    • generate_negative_samples()
    • DataPreprocessor
    • TrainPipeline
    • train_gnn_model()
    • train_kge_model()
    • AttributeDict
    • pickle_dump()
    • pickle_load()
ReDKG
  • redkg package
  • redkg.models package
  • View page source

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()
Previous Next

© Copyright 2022, NCCR Team (ITMO University).

Built with Sphinx using a theme provided by Read the Docs.