config.json

{
    "device": "cpu",
    "dtype": "float32",
    "dimensions": {
        "input": {
            "variables": 2
        },
        "output": {
            "functions": 33,
            "epsilon_orders": 3
        },
        "connection": {
            "epsilon_orders": 3
        }
    },
    "phase_space": {
        "scale": 5.0,
        "const": 1.0,
        "cut": {
            "physical": 0.1,
            "spurious": 0.01
        }
    },
    "hyperparameters": {
        "optimiser": {
            "algorithm": "Adam",
            "initial_learning_rate": 1e-3
        },
        "learning_rate_scheduler": {
            "factor": 0.1,
            "patience": 3,
            "threshold": 1e-2,
            "cooldown": 3
        },
        "iterations": {
            "batch_size": 256,
            "partition_size": 512,
            "number_partitions": 1
        },
        "architecture": {
            "hidden_layers": [128, 128, 128, 128],
            "activation": "GELU"
        },
        "loss_function": {
            "boundary_bias": 1.0,
            "boundary_derivatives": false
        }
    },
    "termination": {
        "epochs": 0,
        "seconds": 0,
        "learning_rate": {
            "target": 1.1e-8,
            "patience": 3
        }
    },
    "zeros": {
        "re": [
            [0, 0],
            [0, 1],
            [1, 0],
            [4, 0],
            [7, 0],
            [8, 0],
            [8, 1],
            [9, 0],
            [9, 1],
            [10, 0],
            [10, 1],
            [11, 0],
            [12, 0],
            [12, 1],
            [14, 0],
            [16, 0],
            [16, 1],
            [19, 0],
            [23, 0],
            [24, 0],
            [25, 0],
            [27, 0]
        ],
        "im": [
            [0, 0],
            [0, 1],
            [1, 0],
            [1, 1],
            [2, 0],
            [3, 0],
            [3, 1],
            [4, 0],
            [4, 1],
            [5, 0],
            [5, 1],
            [6, 0],
            [6, 1],
            [7, 0],
            [8, 0],
            [8, 1],
            [9, 0],
            [9, 1],
            [10, 0],
            [10, 1],
            [11, 0],
            [12, 0],
            [12, 1],
            [13, 0],
            [13, 1],
            [14, 0],
            [14, 1],
            [15, 0],
            [16, 0],
            [16, 1],
            [17, 0],
            [17, 1],
            [18, 0],
            [18, 1],
            [18, 2],
            [19, 0],
            [19, 1],
            [20, 0],
            [20, 1],
            [20, 2],
            [21, 0],
            [21, 1],
            [22, 0],
            [23, 0],
            [23, 1],
            [24, 0],
            [25, 0],
            [25, 1],
            [25, 2],
            [26, 0],
            [26, 1],
            [27, 0],
            [27, 1],
            [28, 0],
            [28, 1],
            [28, 2],
            [29, 0],
            [29, 1],
            [30, 0],
            [30, 1],
            [30, 2],
            [31, 0],
            [32, 0],
            [32, 1],
            [32, 2]
        ]
    }
}