Explorar o código

feat:修改模型训练脚本

zhaohaipeng hai 8 meses
pai
achega
9c738f5b62
Modificáronse 1 ficheiros con 5 adicións e 1 borrados
  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