/MultiOmicsGraphEmbedding

A PyTorch-based package for representation learning on heterogeneous networks. It aims to facilitate experimentation on different GNN models, metrics, and datasets from PytorchGeometric, DGL, networkx.

Primary LanguagePython

MultiOmicsGraphEmbedding

Installation

$ pip install git+https://github.com/JonnyTran/MultiOmicsGraphEmbedding

Package structure:

moge

  • network
  • generator
  • module (PyTorch)
  • model (Tensorflow/Keras)
  • evaluation
  • visualization
  • utils