/GraphNNCenterOfMass

Computing the center of mass of a particle system using Graph Neural Networks

Primary LanguageJupyter Notebook

Computing the center of mass of a particle system using Graph Neural Networks

Create some set of particles with random positions and masses. Implement several Graph Neural Networks models to predict the center of mass of each system with Pytorch Geometric.