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.