分析代码

process_data.py

@@ -118,11 +118,9 @@ if __name__ == "__main__":
     if flag == 1:
         t = "train"
         D = DataProcessor(flag=t)
+        D.producer(dt="whole")
     else:
         t = "predict"
-
-    if flag == 2:
+        D = DataProcessor(flag=t)
         for d in range(16, 22):
             D.producer(d)
-    else:
-        D.producer(dt="whole")