fix bugs

wp8733684 · May 7, 2023 · 20703af · 20703af
1 parent f5d55ff
commit 20703af
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/openstl/datasets/dataloader_taxibj.py b/openstl/datasets/dataloader_taxibj.py
@@ -1,3 +1,4 @@
+import os
 import random
 import numpy as np
 import torch
@@ -42,7 +43,7 @@ def load_data(batch_size, val_batch_size, data_root, num_workers=4,
               pre_seq_length=None, aft_seq_length=None, in_shape=None,
               distributed=False, use_augment=False, use_prefetcher=False):
 
-    dataset = np.load(data_root+'taxibj/dataset.npz')
+    dataset = np.load(os.path.join(data_root, 'taxibj/dataset.npz'))
     X_train, Y_train, X_test, Y_test = dataset['X_train'], dataset[
         'Y_train'], dataset['X_test'], dataset['Y_test']
     assert X_train.shape[1] == pre_seq_length and Y_train.shape[1] == aft_seq_length

diff --git a/openstl/methods/crevnet.py b/openstl/methods/crevnet.py
@@ -18,7 +18,7 @@ class CrevNet(Base_method):
     """
 
     def __init__(self, args, device, steps_per_epoch):
-        args.pre_seq_length = 8
+        args.pre_seq_length = args.pre_seq_length - 2
         args.total_length = args.pre_seq_length + args.aft_seq_length
         Base_method.__init__(self, args, device, steps_per_epoch)
         self.model = self._build_model(self.config)

diff --git a/requirements/docs.txt b/requirements/docs.txt
@@ -8,3 +8,4 @@ sphinx==4.0.2
 sphinx-copybutton
 sphinx_markdown_tables>=0.0.16
 sphinx_rtd_theme==0.5.2
+urllib3==1.26.15