update predict

ZongKe-ZJUT · Nov 16, 2021 · 1c3ffb2 · 1c3ffb2
1 parent f136414
commit 1c3ffb2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data_provider/data_factory.py b/data_provider/data_factory.py
@@ -1,4 +1,4 @@
-from data_provider.data_loader import Dataset_ETT_hour, Dataset_ETT_minute, Dataset_Custom
+from data_provider.data_loader import Dataset_ETT_hour, Dataset_ETT_minute, Dataset_Custom, Dataset_Pred
 from torch.utils.data import DataLoader
 
 data_dict = {
@@ -23,7 +23,7 @@ def data_provider(args, flag):
         shuffle_flag = False
         drop_last = False
         batch_size = 1
-        freq = args.detail_freq
+        freq = args.freq
         Data = Dataset_Pred
     else:
         shuffle_flag = True