Inverted-Pendulum-Neural-Network-Controller/training/initial_conditions.py

from torch import pi

initial_conditions = [
    [1/6 * pi, 0.0, 0.0, 0.0],
    [-1/6 * pi, 0.0, 0.0, 0.0],
    [2/3 * pi, 0.0, 0.0, 0.0],
    [-2/3 * pi, 0.0, 0.0, 0.0],
    [0.0, 1/3 * pi, 0.0, 0.0],
    [0.0, -1/3 * pi, 0.0, 0.0],
    [0.0, 2 * pi, 0.0, 0.0],
    [0.0, -2 * pi, 0.0, 0.0],
    [0.0, 0.0, 0.0, 2 * pi],
    [0.0, 0.0, 0.0, -2 * pi],
    [0.0, 0.0, 0.0, 1/2 * pi],
    [0.0, 0.0, 0.0, -1/2 * pi],
    [0.0, 0.0, 0.0, 1/3 * pi],
    [0.0, 0.0, 0.0, -1/3 * pi],
    [1/4 * pi, 1 * pi, 0.0, 0.0],
    [-1/4 * pi, -1 * pi, 0.0, 0.0],
    [1/2 * pi, -1 * pi, 0.0, 1/3 * pi],
    [-1/2 * pi, 1 * pi, 0.0, -1/3 * pi],
    [1/4 * pi, 1 * pi, 0.0, 2 * pi],
    [-1/4 * pi, -1 * pi, 0.0, 2 * pi],
    [1/2 * pi, -1 * pi, 0.0, 4 * pi],
    [-1/2 * pi, 1 * pi, 0.0, -4 * pi],
]