add full explainer and tsr

khairulislam · Sep 28, 2023 · 87e40bd · 87e40bd
1 parent 1011c27
commit 87e40bd
Show file tree

Hide file tree

Showing 41 changed files with 1,763 additions and 762 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,6 +4,7 @@ lightning_logs
 checkpoints
 # some file sizes are too large for git
 dataset
+*.npy
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

diff --git a/data_provider/data_loader.py b/data_provider/data_loader.py
@@ -51,8 +51,8 @@ def __read_data__(self):
         cols.remove('date')
         df_raw = df_raw[['date'] + cols + [self.target]]
         # print(cols)
-        num_train = int(len(df_raw) * 0.7)
-        num_test = int(len(df_raw) * 0.2)
+        num_train = int(len(df_raw) * 0.8)
+        num_test = int(len(df_raw) * 0.1)
         num_vali = len(df_raw) - num_train - num_test
         border1s = [0, num_train - self.seq_len, len(df_raw) - num_test - self.seq_len]
         border2s = [num_train, num_train + num_vali, len(df_raw)]