特征工程优化

train/train_service.py

@@ -177,10 +177,10 @@ def addDays(n, format="%Y%m%d"):
 
 if __name__ == '__main__':
 
-    trainDays = int(sys.argv[1])
     isTest = False
-    if (len(sys.argv) >= 3 and "test" == sys.argv[2]):
+    if (len(sys.argv) >= 2 and "test" == sys.argv[1]):
         isTest = True
+
     curTime = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
     print("train_service执行时间:{}".format(curTime))
     splitTimestamp = int(time.mktime(time.strptime(addDays(-1), "%Y%m%d")))