3D Charged Particles Dataset - Roto-translated Local Coordinate Frames for Interacting Dynamical Systems

This repository contains the "3D charged particles" dataset from the paper

Roto-translated Local Coordinate Frames for Interacting Dynamical Systems
Miltiadis Kofinas, Naveen Shankar Nagaraja, Efstratios Gavves
NeurIPS 2021
https://arxiv.org/abs/2110.14961
https://github.com/mkofinas/locs

It contains simulations of trajectories of 5 charged particles in 3 dimensions, interacting via Coulomb forces.

There are 30,000 simulations for training, 5,000 for validation, and 5,000 for testing.

Train and validation simulations last for 99 timesteps, while test simulations last for 99 timesteps.

The features comprise positions and velocities of particles, while edges describe the product of pairwise charges.