Jelajahi Sumber

feat:修改模型训练脚本

zhaohaipeng 8 bulan lalu
induk
melakukan
9c738f5b62
1 mengubah file dengan 5 tambahan dan 1 penghapusan
  1. 5 1
      recommend/02_train_go.sh

+ 5 - 1
recommend/02_train_go.sh

@@ -33,7 +33,11 @@ main() {
         elif [ "$train_date" == "$begin_date" ]; then
             $HADOOP fs -text ${hdfs_path}/${train_date}/* | ${FM_TRAIN} -m ${MODEL_PATH}/${model_name}_${train_date}.txt -dim ${train_dim} -core 8
         else
-            yesterday=$(date -d "$train_date -1 day" +%Y%m%d)
+            if [ "$train_date" == 20240801 ]
+                yesterday=$(date -d "$train_date -2 day" +%Y%m%d)
+            else
+                yesterday=$(date -d "$train_date -1 day" +%Y%m%d)
+            fi
             $HADOOP fs -text ${hdfs_path}/${train_date}/* | ${FM_TRAIN} -m ${MODEL_PATH}/${model_name}_${train_date}.txt -dim ${train_dim} -core 8 -im ${MODEL_PATH}/${model_name}_${yesterday}.txt
         fi