config_lstm.json

{
    "paths": {
        "bin_dir": "/data/track-ml/eramia/compiled/",
        "save_dir": "results",
        "log_dir": "logs"
    },
    "data": {
        "filename": "/data/track-ml/eramia/dataset/phi025-025_eta025-025_train1_lasthit_20200219.csv",
        "train_split": 0.80,
        "normalise": true,
        "cylindrical": false,
        "num_hits": 10,
        "features": 3
    },
    "training": {
        "epochs": 21,
        "batch_size": 6,
        "validation": 0.2,
        "save_model": true,
        "load_model": false,
        "use_gpu": true,
        "show_metrics": false,
        "shuffle": true,
        "earlystopping": true
    },
    "testing": {
        "type_optimization" : "normal",
        "normalise": false,
        "tolerance": 0.00
    },
    "model": {
        "name": "lstm",
        "overwrite": true,
        "isparallel": false,                
        "loss": "mse",
        "metrics": ["acc"],
        "optimizer": "adam",
        "layers": [
            {
                "type": "lstm",
                "neurons": 100,
                "input_timesteps": 4,
                "input_features": 3,
                "return_seq": true,
                "stateful": false
            },
            {
                "type": "dropout",
                "rate": 0.25
            },
            {
                "type": "lstm",
                "neurons": 100,
                "input_timesteps": 4,
                "input_features": 3,
                "return_seq": false
            },
            {
                "type": "dense",
                "neurons": 18,
                "activation": "linear"
            },
            {
                "type": "dense",
                "neurons": 3,
                "activation": "linear"
            }
        ]
    }
}