05_update_everyday_ros.sh 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. #!/bin/sh
  2. set -e
  3. set -x
  4. # 0 全局变量/参数
  5. savePath=/dw/recommend/model/ros_sample_v2/
  6. model_name=model_ros_v2
  7. today="$(date +%Y%m%d)"
  8. #yesterday="$(date -d '1 days ago' +%Y%m%d)"
  9. yesterday=20231227
  10. HADOOP="/opt/apps/HADOOP-COMMON/hadoop-common-current/bin/hadoop"
  11. FM_TRAIN="/root/sunmingze/alphaFM/bin/fm_train"
  12. MODEL_PATH="/root/zhangbo/recommend-emr-dataprocess/zhangbo/model/"
  13. OSS_PATH="oss://art-recommend.oss-cn-hangzhou.aliyuncs.com/video_str_model"
  14. # 1 生产数据
  15. #/opt/apps/SPARK2/spark-2.4.8-hadoop3.2-1.0.8/bin/spark-class2 org.apache.spark.deploy.SparkSubmit \
  16. #--class com.aliyun.odps.spark.examples.makedata.makedata_01_readtable2hdfs \
  17. #--master yarn --driver-memory 1G --executor-memory 4G --executor-cores 2 --num-executors 16 \
  18. #/root/zhangbo/recommend-emr-dataprocess/target/spark-examples-1.0.0-SNAPSHOT-shaded.jar \
  19. #tablePart:32 savePath:${savePath} beginStr:${yesterday} endStr:${yesterday}
  20. #
  21. #if [ $? -eq 1 ]; then
  22. # echo "Spark任务执行失败"
  23. # exit 1
  24. #fi
  25. # 2 加载上次模型 训练本轮数据 保存本轮模型
  26. end_date=${today}
  27. loop_date=${yesterday}
  28. while [[ "$loop_date" != "$end_date" ]]; do
  29. echo -------train ${loop_date}----------
  30. loop_date_model=$(date -d "$loop_date - 1 day" +%Y%m%d)
  31. $HADOOP fs -text ${savePath}/dt=${loop_date}/* | ${FM_TRAIN} -m ${MODEL_PATH}/${model_name}_${loop_date}.txt \
  32. -dim 1,1,0 -core 8 -im ${MODEL_PATH}/${model_name}_${loop_date_model}.txt
  33. if [ $? -eq 1 ]; then
  34. echo "训练失败"
  35. exit 1
  36. fi
  37. echo -------save ${MODEL_PATH}/${model_name}_${loop_date}.txt----------
  38. loop_date=$(date -d "$loop_date + 1 day" +%Y%m%d)
  39. done
  40. # 3 本轮模型格式转换
  41. cat ${MODEL_PATH}/${model_name}_${yesterday}.txt \
  42. | sed '1d' | awk -F " " '{if($2!="0") print $1"\t"$2}' \
  43. > ${MODEL_PATH}/${model_name}_${yesterday}_change.txt
  44. # 4 转换后模型上传oss
  45. $HADOOP fs -rm -r oss://art-recommend.oss-cn-hangzhou.aliyuncs.com/video_str_model/${model_name}_change.txt
  46. $HADOOP fs -put ${MODEL_PATH}/${model_name}_${yesterday}_change.txt ${OSS_PATH}/${model_name}_change.txt
  47. #nohup sh 05_update_everyday_ros.sh > p.log 2>&1 &