/PyTorchGeometric_Demo

Primary LanguagePythonMIT LicenseMIT

PyTorchGeometric_Demo

This is a demo showing how to do representation learning for sequence data use PyTorch Geometric library. Each sequence will be learned as a weighted graph.

Environment

  • Python 3.7.6
  • PyTorch 1.4.0
  • PyTorch Geometric 1.6.0