jiongWs commited on
Commit
2c41964
·
1 Parent(s): 82db2a5

add config

Browse files
Files changed (1) hide show
  1. config.json +440 -0
config.json ADDED
@@ -0,0 +1,440 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataloader": {
3
+ "kwargs": {
4
+ "batch_size": 2048
5
+ },
6
+ "type": "pytorch_dataloader"
7
+ },
8
+ "dataset": {
9
+ "kwargs": {
10
+ "feature_columns": [
11
+ "USM5",
12
+ "USM5inday",
13
+ "MA60",
14
+ "MA60inday",
15
+ "QTLD60",
16
+ "QTLD60inday",
17
+ "RM30",
18
+ "BSRL30",
19
+ "BSRL30inday",
20
+ "MIN30",
21
+ "MIN30inday",
22
+ "QTLD30",
23
+ "QTLD30inday",
24
+ "MA30",
25
+ "MA30inday",
26
+ "BETA10",
27
+ "BETA10inday",
28
+ "MIN10",
29
+ "MIN10inday",
30
+ "ROC30",
31
+ "ROC30inday",
32
+ "ESR5",
33
+ "ESR5inday",
34
+ "BSRL5",
35
+ "BSRL5inday",
36
+ "USV5",
37
+ "RM5",
38
+ "RM5inday",
39
+ "MIN20",
40
+ "USM10",
41
+ "USM10inday",
42
+ "CNTP5",
43
+ "CNTP5inday",
44
+ "USM30",
45
+ "USM30inday",
46
+ "SUMP30",
47
+ "SUMP30inday",
48
+ "SUMD30",
49
+ "SUMD30inday",
50
+ "SUMN30",
51
+ "SUMN30inday",
52
+ "SUMP5",
53
+ "SUMP5inday",
54
+ "SUMD5",
55
+ "SUMD5inday",
56
+ "SUMN5",
57
+ "USM20",
58
+ "USM20inday",
59
+ "CNTD5",
60
+ "CNTD5inday",
61
+ "MIN60",
62
+ "MIN60inday",
63
+ "ROC5",
64
+ "ROC5inday",
65
+ "ESR30",
66
+ "ESR30inday",
67
+ "MA20",
68
+ "MA20inday",
69
+ "QTLD20",
70
+ "QTLD20inday",
71
+ "RM20",
72
+ "RM20inday",
73
+ "QTLU60",
74
+ "QTLU60inday",
75
+ "UMDSV30",
76
+ "UMDSV30inday",
77
+ "QTLD10",
78
+ "QTLD10inday",
79
+ "RM60",
80
+ "RM60inday",
81
+ "MA10",
82
+ "MA10inday",
83
+ "CNTN5",
84
+ "CNTN5inday",
85
+ "USV30",
86
+ "USV30inday",
87
+ "QTLU30",
88
+ "QTLU30inday",
89
+ "MIN5",
90
+ "MIN5inday",
91
+ "RM10",
92
+ "RM10inday",
93
+ "ONRM10",
94
+ "ONRM10inday",
95
+ "ROC20inday",
96
+ "USV20",
97
+ "USV20inday",
98
+ "SUMP60",
99
+ "SUMP60inday",
100
+ "SUMD60",
101
+ "SUMD60inday",
102
+ "SUMN60",
103
+ "SUMN60inday",
104
+ "RSV30",
105
+ "KLEN",
106
+ "KLENinday",
107
+ "USV10",
108
+ "USV10inday",
109
+ "ESR60",
110
+ "ESR60inday",
111
+ "UMDSV5",
112
+ "UMDSV5inday",
113
+ "UMDSV60",
114
+ "UMDSV60inday",
115
+ "ROC60",
116
+ "ROC60inday",
117
+ "RANK60",
118
+ "RANK60inday",
119
+ "ROC10",
120
+ "ROC10inday",
121
+ "ESR10inday",
122
+ "ESR20",
123
+ "ESR20inday",
124
+ "SUMP10",
125
+ "SUMP10inday",
126
+ "SUMD10",
127
+ "SUMD10inday",
128
+ "SUMN10",
129
+ "SUMN10inday",
130
+ "SUMP20",
131
+ "SUMD20",
132
+ "SUMD20inday",
133
+ "SUMN20",
134
+ "SUMN20inday",
135
+ "BETA30",
136
+ "BETA30inday",
137
+ "IMXD5",
138
+ "IMXD5inday",
139
+ "USM60",
140
+ "USM60inday",
141
+ "CNTP10",
142
+ "CNTP10inday",
143
+ "QTLU20",
144
+ "IMXD10",
145
+ "IMXD10inday",
146
+ "USV60",
147
+ "USV60inday",
148
+ "CNTD10",
149
+ "RANK30",
150
+ "RANK30inday",
151
+ "RSV20",
152
+ "RSV20inday",
153
+ "CNTD30",
154
+ "CNTD30inday",
155
+ "RSV10",
156
+ "RSV10inday",
157
+ "IMAX5",
158
+ "IMAX5inday",
159
+ "UMDSV20",
160
+ "UMDSV20inday",
161
+ "BETA5inday",
162
+ "CNTP30",
163
+ "CNTP30inday",
164
+ "BETA60",
165
+ "BETA60inday",
166
+ "RANK10",
167
+ "RANK10inday",
168
+ "IMAX10",
169
+ "IMAX10inday",
170
+ "RANK20",
171
+ "RANK20inday",
172
+ "CNTN10",
173
+ "CNTN10inday",
174
+ "UMDSV10",
175
+ "UMDSV10inday",
176
+ "CNTP20",
177
+ "CNTP20inday",
178
+ "CNTD20",
179
+ "CNTD20inday",
180
+ "QTLU10inday",
181
+ "UMDSM5",
182
+ "UMDSM5inday",
183
+ "UMDSM20",
184
+ "UMDSM20inday",
185
+ "CNTN30",
186
+ "CNTN30inday",
187
+ "IMXD30",
188
+ "IMXD30inday",
189
+ "BETA20",
190
+ "BETA20inday",
191
+ "BSRL20",
192
+ "BSRL20inday",
193
+ "UMDSM30",
194
+ "UMDSM30inday",
195
+ "UMDSM10",
196
+ "MAX60",
197
+ "MAX60inday",
198
+ "IMAX30",
199
+ "IMAX30inday",
200
+ "CNTD60",
201
+ "CNTD60inday",
202
+ "IMXD60",
203
+ "IMXD60inday",
204
+ "IMAX60",
205
+ "IMAX60inday",
206
+ "CNTN20",
207
+ "CNTN20inday",
208
+ "BSRL10",
209
+ "BSRL10inday",
210
+ "CORD5",
211
+ "CORD5inday",
212
+ "IMIN10",
213
+ "IMIN10inday",
214
+ "CNTP60",
215
+ "CNTP60inday",
216
+ "CORD30",
217
+ "CORD30inday",
218
+ "LOW0",
219
+ "LOW0inday",
220
+ "MAX30",
221
+ "MAX30inday",
222
+ "IMIN5",
223
+ "IMIN5inday",
224
+ "VRMIN30",
225
+ "VSUMP60",
226
+ "VSUMP60inday",
227
+ "VSUMN60",
228
+ "VSUMN60inday",
229
+ "VSUMD60",
230
+ "VSUMD60inday",
231
+ "KLOW",
232
+ "KLOWinday",
233
+ "UMDSM60",
234
+ "UMDSM60inday",
235
+ "RESI60",
236
+ "CNTN60",
237
+ "VRMIN10",
238
+ "VRMIN10inday",
239
+ "RSV5",
240
+ "RSV5inday",
241
+ "IMXD20",
242
+ "IMXD20inday",
243
+ "VRMIN20inday",
244
+ "VRMAX5",
245
+ "VRMAX5inday",
246
+ "DSM5",
247
+ "DSM5inday",
248
+ "VSTD20",
249
+ "VSTD20inday",
250
+ "DSV60",
251
+ "DSV60inday",
252
+ "VSTD10",
253
+ "VSTD10inday",
254
+ "KSFT2",
255
+ "KSFT2inday",
256
+ "DSV20",
257
+ "DSV20inday",
258
+ "RS30",
259
+ "RS30inday",
260
+ "WVMA10",
261
+ "WVMA10inday",
262
+ "MAX5",
263
+ "MAX5inday",
264
+ "EMDD5",
265
+ "EMDD5inday",
266
+ "KMID2",
267
+ "KMID2inday",
268
+ "STD20",
269
+ "STD20inday",
270
+ "WVMA20",
271
+ "WVMA20inday",
272
+ "DSV30",
273
+ "DSV30inday",
274
+ "WVMA5",
275
+ "WVMA5inday",
276
+ "EMDD20",
277
+ "EMDD20inday",
278
+ "CORR5",
279
+ "CORR5inday",
280
+ "WVMA30",
281
+ "WVMA30inday",
282
+ "WVMA60",
283
+ "WVMA60inday",
284
+ "DSM10",
285
+ "DSM10inday",
286
+ "KUP2",
287
+ "KUP2inday",
288
+ "DSV5",
289
+ "DSV5inday",
290
+ "KMID",
291
+ "KMIDinday",
292
+ "KSFT",
293
+ "KSFTinday",
294
+ "ONRM30",
295
+ "STD30",
296
+ "STD30inday",
297
+ "VSTD5",
298
+ "VSTD5inday",
299
+ "DSV10",
300
+ "DSV10inday",
301
+ "RSQR30",
302
+ "RSQR30inday",
303
+ "RSQR60",
304
+ "RSQR60inday",
305
+ "OPEN0",
306
+ "OPEN0inday",
307
+ "DSM60",
308
+ "DSM60inday",
309
+ "RS20",
310
+ "RS20inday",
311
+ "RSQR20",
312
+ "RSQR20inday",
313
+ "RK20",
314
+ "RK20inday",
315
+ "DB30",
316
+ "DB30inday",
317
+ "RK30",
318
+ "RK30inday",
319
+ "ONRM60",
320
+ "DB10",
321
+ "DB10inday",
322
+ "ONRM20",
323
+ "ONRM20inday",
324
+ "RSQR10",
325
+ "RSQR10inday",
326
+ "STD60inday",
327
+ "RSQR5",
328
+ "RSQR5inday",
329
+ "KLOW2",
330
+ "KLOW2inday",
331
+ "DB20",
332
+ "DB20inday",
333
+ "DSM30",
334
+ "DSM30inday",
335
+ "RESI10",
336
+ "RESI10inday",
337
+ "DSM20",
338
+ "DSM20inday",
339
+ "ONRM5",
340
+ "ONRM5inday",
341
+ "DB5",
342
+ "DB5inday"
343
+ ],
344
+ "label_columns": [
345
+ "label1"
346
+ ],
347
+ "mask": {
348
+ "kwargs": {},
349
+ "type": "pad_mask"
350
+ }
351
+ },
352
+ "type": "TransformerDatasetWithDict"
353
+ },
354
+ "feature_file": "configs/low_corr.txt",
355
+ "loss": {
356
+ "kwargs": {
357
+ "mean_n": 0.0,
358
+ "std_n": 0.0,
359
+ "weight_change": [
360
+ [
361
+ 0.05,
362
+ 0.3
363
+ ]
364
+ ]
365
+ },
366
+ "type": "CorrLoss"
367
+ },
368
+ "max_epochs": 20,
369
+ "min_epochs": 10,
370
+ "model": {
371
+ "kwargs": {
372
+ "embedding_config": {
373
+ "kwargs": {
374
+ "dropout": 0.2,
375
+ "fixed": false,
376
+ "inlinear": true
377
+ },
378
+ "type": "Exp2dEmbedding"
379
+ },
380
+ "embedding_size": 256,
381
+ "encoder_config": {
382
+ "kwargs": {
383
+ "activation": "gelu",
384
+ "d_ff": 256,
385
+ "d_k": 64,
386
+ "d_v": 64,
387
+ "layer_num": 4,
388
+ "n_heads": 16,
389
+ "res_dropout": 0.1
390
+ },
391
+ "type": "TSTPlusEncoder"
392
+ },
393
+ "feature_size": 256,
394
+ "head_config": {
395
+ "kwargs": {
396
+ "dropout": 0.5,
397
+ "hidden_size": 256,
398
+ "num_layers": 1
399
+ },
400
+ "type": "FCHead"
401
+ },
402
+ "input_size": 332,
403
+ "pooling_config": {
404
+ "kwargs": {},
405
+ "type": "LastPooling"
406
+ },
407
+ "y_len": 1
408
+ },
409
+ "type": "TransformerModel"
410
+ },
411
+ "model_name": "tst_t1_base_20_2",
412
+ "monitor": "val_loss",
413
+ "norm_mean": 0.0676409454581534,
414
+ "norm_std": 2.1221225816121017,
415
+ "optimizer": {
416
+ "kwargs": {
417
+ "lr": 5e-05,
418
+ "weight_decay": 1e-06
419
+ },
420
+ "type": "Adam"
421
+ },
422
+ "patience": 10,
423
+ "pp": "t1_0",
424
+ "scheduler": {
425
+ "kwargs": {
426
+ "gamma": 0.5,
427
+ "milestones": [
428
+ 5,
429
+ 15
430
+ ]
431
+ },
432
+ "type": "MultiStepLR"
433
+ },
434
+ "seq_length": 20,
435
+ "test_dataset": "dataset/test_000300_t1_2101-2107.csv",
436
+ "train_dataset": "dataset/train_000906_t1_0801-1901.csv",
437
+ "transformers_version": "4.10.0.dev0",
438
+ "use_cache": true,
439
+ "valid_dataset": "dataset/valid_000300_t1_1901-2101.csv"
440
+ }