changed to default model as glstm

youngjoo-epfl · Oct 10, 2017 · 816d7b5 · 816d7b5
1 parent 8b8f80b
commit 816d7b5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/config.py b/config.py
@@ -41,7 +41,7 @@ def add_argument_group(name):
 
 # Model args
 model_arg = add_argument_group('Model')
-model_arg.add_argument('--model_type', type=str, default='lstm',
+model_arg.add_argument('--model_type', type=str, default='glstm',
                         choices=['lstm', 'glstm'], help='')
 
 # Hyperparams for graph

diff --git a/utils.py b/utils.py
@@ -85,10 +85,10 @@ def __init__(self, data_dir, dataset_name, batch_size, seq_length):
         print("Reshaping tensors...")
         for split, data in enumerate(all_data):  # split = 0:train, 1:valid, 2:test
             #Cutting training sample for check profile fast..(Temporal)
-            if split==0:
-                #Onlu for training set
-                length = data.shape[0]
-                data = data[:int(length/4)]
+            #if split==0:
+            #    #Only for training set
+            #    length = data.shape[0]
+            #    data = data[:int(length/4)]
 
             length = data.shape[0]
             data = data[: batch_size * seq_length * int(math.floor(length / (batch_size * seq_length)))]