current_model_set.json

{
    "DAGMM": {
        "toTrain": true,
        "toPredict": true,
        "data_path": "data/June30_mars_4min_an",
        "weight_path": "weights/DAGMM",
        "predictions_path": "predictions/DAGMM",
        "train_begin": 0,
        "train_end": 10455,
        "test_begin": 0,
        "test_end": 10470,
        "forecast_length": 15,
        "skip_length": 0,
        "training_params": {
            "lambda_energy": 0.2,
            "lr": 0.001,
            "batch_size": 50,
            "gmm_k": 3,
            "num_epochs": 5,
            "sequence_length": 20
        }
    },
    "LSTM_DAGMM": {
        "toTrain": false,
        "toPredict": false,
        "data_path": "data/June30_mars_4min_an",
        "weight_path": "weights/LSTM_DAGMM",
        "predictions_path": "predictions/LSTM_DAGMM",
        "train_begin": 0,
        "train_end": 10800,
        "test_begin": 0,
        "test_end": null,
        "training_params": {
            "lambda_energy": 0.2,
            "lr": 0.001,
            "batch_size": 50,
            "gmm_k": 3,
            "num_epochs": 1,
            "sequence_length": 20
        }
    }
}