Quellcode durchsuchen

feat:修改模型训练脚本

zhaohaipeng vor 8 Monaten
Ursprung
Commit
9c738f5b62
1 geänderte Dateien mit 5 neuen und 1 gelöschten Zeilen
  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