瀏覽代碼

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

zhaohaipeng 8 月之前
父節點
當前提交
4390cbbf5e
共有 1 個文件被更改,包括 3 次插入1 次删除
  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