first commit

chl8856 · Jun 10, 2020 · bf7b3e7 · bf7b3e7
commit bf7b3e7
Show file tree

Hide file tree

Showing 10 changed files with 4,500 additions and 0 deletions.
diff --git a/class_AC_TPC.py b/class_AC_TPC.py
diff --git a/data/.ipynb_checkpoints/data_ADNI_preprocess-checkpoint.ipynb b/data/.ipynb_checkpoints/data_ADNI_preprocess-checkpoint.ipynb
diff --git a/data/.ipynb_checkpoints/data_CF_label_test-checkpoint.ipynb b/data/.ipynb_checkpoints/data_CF_label_test-checkpoint.ipynb
diff --git a/data/.ipynb_checkpoints/data_CF_preprocess-checkpoint.ipynb b/data/.ipynb_checkpoints/data_CF_preprocess-checkpoint.ipynb
diff --git a/data/sample/data.npz b/data/sample/data.npz
diff --git a/data_loader.py b/data_loader.py
@@ -0,0 +1,19 @@
+import numpy as np
+
+
+def import_data(data_name = 'sample'):
+    '''
+        Output: 
+            - data_x: [N, max_length, 1+x_dim] tensor (where N: number of samples, max_length: max sequence length, x_dim: feature dimension)
+                      the first feature is the time difference.
+            - data_y: [N, max_length, y_dim] tensor (where N: number of samples, max_length: max sequence length, y_dim: output dimension)
+    '''
+    if data_name == 'sample':
+        npz    = np.load('./data/sample/data.npz')
+        data_x = npz['data_x']
+        data_y = npz['data_y']
+        y_type = npz['y_type'] #{'binary', 'categorical', 'continuous'}
+    else:
+        raise ValueError('error: data_name not defined')
+
+    return data_x, data_y, y_type