Bläddra i källkod

feat:添加推荐模型新模型训练脚本

zhaohaipeng 8 månader sedan
förälder
incheckning
4390cbbf5e
1 ändrade filer med 3 tillägg och 1 borttagningar
  1. 3 1
      recommend/02_train_go.sh

+ 3 - 1
recommend/02_train_go.sh

@@ -19,11 +19,13 @@ FM_TRAIN=/root/sunmingze/alphaFM/bin/fm_train
 
 
 train_date=$begin_date
+train_end_time=$(date -d "$end_date +1 day" +%Y%m%d)
 
 main() {
 
     # 增量训练模型
-    while [ "$train_date" != "$end_date" ]; do
+
+    while [ "$train_date" != "$train_end_time" ]; do
         echo "==================== 开始训练 $train_date 模型 ===================="
 
         if [ "$train_date" == "$begin_date" ]; then