Published January 22, 2025 | Version v1
Dataset Open

CODES Dataset: coupled_oscillators

Authors/Creators

  • 1. EDMO icon University of Heidelberg

Description

Toy dataset for the CODES benchmark (https://github.com/robin-janssen/CODES-Benchmark).

Five harmonic oscillators, coupled with linear and non-linear springs. Each oscillator has a position x_i and a velocity v_i.

The following system of equations describes the dynamics:

Position derivatives
dx1_dt = v1
dx2_dt = v2
dx3_dt = v3
dx4_dt = v4
dx5_dt = v5 

Velocity derivatives
dv1_dt = -0.1 * v1 - 2.0 * (x1 - x2) - 0.5 * (x1 - x2)**30.05 * (x3 - x1)**2
dv2_dt = -0.1 * v2 - 2.0 * (x2 - x1) - 0.5 * (x2 - x1)**32.0 * (x2 - x3) - 0.5 * (x2 - x3)**30.05 * (x4 - x2)**2
dv3_dt = -0.1 * v3 - 2.0 * (x3 - x2) - 0.5 * (x3 - x2)**32.0 * (x3 - x4) - 0.5 * (x3 - x4)**30.05 * (x5 - x3)**2
dv4_dt = -0.1 * v4 - 2.0 * (x4 - x3) - 0.5 * (x4 - x3)**32.0 * (x4 - x5) - 0.5 * (x4 - x5)**30.05 * (x1 - x4)**2
dv5_dt = -0.1 * v5 - 2.0 * (x5 - x4) - 0.5 * (x5 - x4)**30.05 * (x2 - x5)**2

101 timesteps in the interval [0, 3], solved with scipy.integrate.solve_ivp with the DOP853 method (atol = rtol = 1e-8).
Initial conditions for each species sobol-sampled in the interval (-1, 1). 
 
700/100/200 train/test/val samples.

Files

Files (8.1 MB)

Name Size Download all
md5:149019fd392dadab4b3cf960757be35d
8.1 MB Download

Additional details

Software

Repository URL
https://github.com/robin-janssen/CODES-Benchmark
Programming language
Python
Development Status
Active