训练数据时间更改

spark/featureEng.py

@@ -709,7 +709,7 @@ if __name__ == '__main__':
     train_columns = model_columns + ["label", "timestamp"]
     trainSamples = samplesWithUserFeatures.select(*train_columns)
     print("write to hdfs start...")
-    splitTimestamp = int(time.mktime(time.strptime(endDay, "%Y%m%d")))
+    splitTimestamp = int(time.mktime(time.strptime(addDays(0), "%Y%m%d")))
     splitAndSaveTrainingTestSamplesByTimeStamp(trainSamples, splitTimestamp, TRAIN_FILE_PATH)
     print("write to hdfs success...")
     timestmp7 = int(round(time.time()))