Timeband/model.cfg at develop · handal95/Timeband · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
{
    "train_mode": true,
    "clean_mode": true,
    "preds_mode": false,
    "core": {
        "TAG": "refactor",
        "data_name": "000060",
        "directory": "data/",
        "pretrain": true,
        "workers": 2,
        "batch_size": 1024,
        "observed_len": 10,
        "forecast_len": 3,
        "verbosity": 0
    },
    "dataset": {
        "reset": true,
        "targets": ["Close"],
        "drops": [],
        "time_index": "Date",
        "time_format": "%Y-%m-%d",
        "time_encode": {
            "year": true,
            "month": true,
            "weekday": false,
            "day": true,
            "hour": true,
            "minute": false
        },
        "fill_timegap": false,
        "stride": 1,
        "split_rate": 0.99,
        "min_valid_scale": 2,
        "sliding_step": 0
    },
    "models": {
        "save_opt": true,
        "load_opt": true,
        "best_score": 50.0,
        "hidden_dim": 256,
        "layers_num": 1
    },
    "losses": {
        "l1_weight": 50,
        "l2_weight": 50,
        "gp_weight": 50
    },
    "trainer": {
        "models": {
            "reload": false,
            "reload_interval": 50,
            "save_interval": 10
        },
        "critic": 1,
        "epochs": 50,
        "learning_rate": {
            "base": 2e-4,
            "decay": 0.9,
            "gammaD": 1.0,
            "gammaG": 1.0
        },
        "band_width": 1,
        "missing_gamma": 0.2,
        "anomaly_gamma": 0.0,
        "print_interval": 5
    },
    "dashboard": {
        "vis_opt": false,
        "vis_column": 1,
        "visual_cols": [
            "21"
        ],
        "scope": 120,
        "feats_by_rows": 1,
        "xinterval": 14,
        "height": 10,
        "width": 25
    }
}