add config

config.json

@@ -0,0 +1,447 @@
+{
+  "dataloader": {
+    "kwargs": {
+      "batch_size": 2048
+    },
+    "type": "pytorch_dataloader"
+  },
+  "dataset": {
+    "kwargs": {
+      "feature_columns": [
+        "USM5",
+        "USM5inday",
+        "MA60",
+        "MA60inday",
+        "QTLD60inday",
+        "RM30",
+        "RM30inday",
+        "BSRL30",
+        "BSRL30inday",
+        "MIN30",
+        "MIN30inday",
+        "QTLD30",
+        "QTLD30inday",
+        "MA30",
+        "MA30inday",
+        "BETA10",
+        "BETA10inday",
+        "MIN10",
+        "MIN10inday",
+        "ROC30",
+        "ROC30inday",
+        "ESR5",
+        "ESR5inday",
+        "BSRL5",
+        "BSRL5inday",
+        "USV5",
+        "USV5inday",
+        "RM5",
+        "RM5inday",
+        "MIN20",
+        "MIN20inday",
+        "USM10",
+        "USM10inday",
+        "CNTP5",
+        "CNTP5inday",
+        "USM30",
+        "USM30inday",
+        "SUMP30",
+        "SUMP30inday",
+        "SUMD30",
+        "SUMD30inday",
+        "SUMN30",
+        "SUMN30inday",
+        "SUMP5",
+        "SUMP5inday",
+        "SUMD5",
+        "SUMN5",
+        "SUMN5inday",
+        "USM20",
+        "USM20inday",
+        "CNTD5",
+        "CNTD5inday",
+        "MIN60",
+        "MIN60inday",
+        "ROC5",
+        "ROC5inday",
+        "ESR30",
+        "ESR30inday",
+        "RSV60",
+        "RSV60inday",
+        "MA20",
+        "MA20inday",
+        "QTLD20",
+        "QTLD20inday",
+        "RM20",
+        "QTLU60",
+        "QTLU60inday",
+        "UMDSV30",
+        "UMDSV30inday",
+        "QTLD10",
+        "QTLD10inday",
+        "RM60",
+        "RM60inday",
+        "MA10",
+        "MA10inday",
+        "CNTN5",
+        "CNTN5inday",
+        "USV30",
+        "USV30inday",
+        "QTLU30",
+        "QTLU30inday",
+        "MIN5",
+        "MIN5inday",
+        "RM10",
+        "RM10inday",
+        "ONRM10",
+        "ONRM10inday",
+        "ROC20",
+        "ROC20inday",
+        "USV20",
+        "USV20inday",
+        "SUMP60",
+        "SUMP60inday",
+        "SUMD60",
+        "SUMN60",
+        "RSV30",
+        "RSV30inday",
+        "KLEN",
+        "KLENinday",
+        "USV10",
+        "USV10inday",
+        "ESR60",
+        "ESR60inday",
+        "UMDSV5",
+        "UMDSV5inday",
+        "UMDSV60",
+        "ROC60",
+        "ROC60inday",
+        "RANK60",
+        "RANK60inday",
+        "ROC10",
+        "ROC10inday",
+        "ESR10",
+        "ESR10inday",
+        "ESR20",
+        "ESR20inday",
+        "SUMP10",
+        "SUMP10inday",
+        "SUMD10",
+        "SUMD10inday",
+        "SUMN10",
+        "SUMN10inday",
+        "SUMP20",
+        "SUMP20inday",
+        "SUMD20",
+        "SUMD20inday",
+        "SUMN20",
+        "SUMN20inday",
+        "BETA30",
+        "BETA30inday",
+        "IMXD5",
+        "IMXD5inday",
+        "USM60",
+        "USM60inday",
+        "CNTP10",
+        "CNTP10inday",
+        "QTLU20",
+        "QTLU20inday",
+        "IMXD10",
+        "IMXD10inday",
+        "USV60",
+        "USV60inday",
+        "CNTD10",
+        "CNTD10inday",
+        "RANK30",
+        "RANK30inday",
+        "RSV20",
+        "RSV20inday",
+        "CNTD30",
+        "CNTD30inday",
+        "RSV10",
+        "RSV10inday",
+        "IMAX5",
+        "IMAX5inday",
+        "UMDSV20",
+        "UMDSV20inday",
+        "BETA5",
+        "BETA5inday",
+        "CNTP30",
+        "CNTP30inday",
+        "BETA60",
+        "BETA60inday",
+        "RANK10",
+        "RANK10inday",
+        "IMAX10",
+        "IMAX10inday",
+        "RANK20",
+        "RANK20inday",
+        "CNTN10",
+        "CNTN10inday",
+        "UMDSV10",
+        "UMDSV10inday",
+        "CNTP20",
+        "CNTP20inday",
+        "CNTD20inday",
+        "QTLU10",
+        "QTLU10inday",
+        "UMDSM5",
+        "UMDSM5inday",
+        "UMDSM20",
+        "UMDSM20inday",
+        "CNTN30",
+        "CNTN30inday",
+        "IMXD30",
+        "IMXD30inday",
+        "BETA20",
+        "BETA20inday",
+        "BSRL20",
+        "BSRL20inday",
+        "UMDSM30",
+        "UMDSM30inday",
+        "UMDSM10",
+        "UMDSM10inday",
+        "MAX60inday",
+        "IMAX30",
+        "IMAX30inday",
+        "CNTD60",
+        "CNTD60inday",
+        "IMXD60",
+        "IMXD60inday",
+        "IMAX60",
+        "CNTN20",
+        "CNTN20inday",
+        "BSRL10",
+        "BSRL10inday",
+        "CORD5",
+        "CORD5inday",
+        "IMIN10",
+        "IMIN10inday",
+        "CNTP60",
+        "CNTP60inday",
+        "CORD30",
+        "CORD30inday",
+        "LOW0",
+        "LOW0inday",
+        "MAX30",
+        "MAX30inday",
+        "IMIN5",
+        "IMIN5inday",
+        "VRMIN30",
+        "VRMIN30inday",
+        "VSUMP60",
+        "VSUMP60inday",
+        "VSUMN60",
+        "VSUMN60inday",
+        "VSUMD60",
+        "VSUMD60inday",
+        "KLOW",
+        "KLOWinday",
+        "UMDSM60inday",
+        "RESI60",
+        "RESI60inday",
+        "CNTN60",
+        "CNTN60inday",
+        "VRMIN10",
+        "VRMIN10inday",
+        "RSV5",
+        "RSV5inday",
+        "IMXD20",
+        "IMXD20inday",
+        "VRMIN20",
+        "VRMIN20inday",
+        "VRMAX5",
+        "VRMAX5inday",
+        "DSM5",
+        "DSM5inday",
+        "VSTD20",
+        "VSTD20inday",
+        "DSV60",
+        "DSV60inday",
+        "VSTD10",
+        "VSTD10inday",
+        "KSFT2",
+        "KSFT2inday",
+        "DSV20",
+        "RS30",
+        "RS30inday",
+        "WVMA10",
+        "MAX5",
+        "MAX5inday",
+        "EMDD5",
+        "EMDD5inday",
+        "KMID2",
+        "KMID2inday",
+        "STD20",
+        "WVMA20",
+        "WVMA20inday",
+        "DSV30",
+        "DSV30inday",
+        "WVMA5",
+        "WVMA5inday",
+        "EMDD20",
+        "CORR5",
+        "CORR5inday",
+        "WVMA30",
+        "WVMA30inday",
+        "WVMA60",
+        "WVMA60inday",
+        "DSM10",
+        "DSM10inday",
+        "KUP2",
+        "KUP2inday",
+        "DSV5",
+        "DSV5inday",
+        "KMID",
+        "KMIDinday",
+        "KSFT",
+        "KSFTinday",
+        "ONRM30inday",
+        "STD30",
+        "STD30inday",
+        "VSTD5",
+        "VSTD5inday",
+        "DSV10",
+        "DSV10inday",
+        "RSQR30",
+        "RSQR30inday",
+        "RSQR60",
+        "RSQR60inday",
+        "OPEN0",
+        "OPEN0inday",
+        "DSM60",
+        "DSM60inday",
+        "RS20",
+        "RS20inday",
+        "RSQR20",
+        "RSQR20inday",
+        "RK20",
+        "RK20inday",
+        "DB30",
+        "DB30inday",
+        "RK30",
+        "RK30inday",
+        "ONRM60",
+        "ONRM60inday",
+        "DB10",
+        "DB10inday",
+        "ONRM20",
+        "ONRM20inday",
+        "RSQR10",
+        "RSQR10inday",
+        "STD60",
+        "STD60inday",
+        "RSQR5",
+        "RSQR5inday",
+        "KLOW2",
+        "KLOW2inday",
+        "DB20",
+        "DB20inday",
+        "DSM30",
+        "DSM30inday",
+        "RESI10",
+        "RESI10inday",
+        "DSM20",
+        "DSM20inday",
+        "ONRM5",
+        "ONRM5inday",
+        "DB5",
+        "DB5inday"
+      ],
+      "label_columns": [
+        "label1"
+      ],
+      "mask": {
+        "kwargs": {},
+        "type": "pad_mask"
+      }
+    },
+    "type": "TransformerDatasetWithDict"
+  },
+  "feature_file": "configs/low_corr.txt",
+  "loss": {
+    "kwargs": {
+      "mean_n": 0.0,
+      "std_n": 0.0,
+      "weight_change": [
+        [
+          0.05,
+          0.3
+        ]
+      ]
+    },
+    "type": "CorrLoss"
+  },
+  "max_epochs": 20,
+  "min_epochs": 10,
+  "model": {
+    "kwargs": {
+      "embedding_config": {
+        "kwargs": {
+          "dropout": 0.2,
+          "fixed": false,
+          "inlinear": true
+        },
+        "type": "Exp2dEmbedding"
+      },
+      "embedding_size": 256,
+      "encoder_config": {
+        "kwargs": {
+          "activation": "gelu",
+          "d_ff": 256,
+          "d_k": 64,
+          "d_v": 64,
+          "layer_num": 4,
+          "n_heads": 16,
+          "res_dropout": 0.1
+        },
+        "type": "TSTPlusEncoder"
+      },
+      "feature_size": 256,
+      "head_config": {
+        "kwargs": {
+          "dropout": 0.5,
+          "hidden_size": 256,
+          "num_layers": 1
+        },
+        "type": "FCHead"
+      },
+      "input_size": 339,
+      "pooling_config": {
+        "kwargs": {},
+        "type": "LastPooling"
+      },
+      "y_len": 1
+    },
+    "type": "TransformerModel"
+  },
+  "model_name": "tst_t1_base_20_1",
+  "monitor": "val_loss",
+  "norm_mean": 0.03223681614149798,
+  "norm_std": 2.1309957578431993,
+  "optimizer": {
+    "kwargs": {
+      "lr": 5e-05,
+      "weight_decay": 1e-06
+    },
+    "type": "Adam"
+  },
+  "patience": 10,
+  "pp": "t1_0",
+  "scheduler": {
+    "kwargs": {
+      "gamma": 0.5,
+      "milestones": [
+        5,
+        15
+      ]
+    },
+    "type": "MultiStepLR"
+  },
+  "seq_length": 20,
+  "test_dataset": "dataset/test_000300_t1_2101-2107.csv",
+  "train_dataset": "dataset/train_000906_t1_0801-1901.csv",
+  "transformers_version": "4.10.0.dev0",
+  "use_cache": true,
+  "valid_dataset": "dataset/valid_000300_t1_1901-2101.csv"
+}