handle_rov.sh 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. #!/bin/sh
  2. set -ex
  3. source /root/anaconda3/bin/activate py37
  4. # nohup sh handle_rov.sh > "$(date +%Y%m%d_%H%M%S)_handle_rov.log" 2>&1 &
  5. # 原始数据table name
  6. table='alg_recsys_sample_all'
  7. today="$(date +%Y%m%d)"
  8. today_early_3="$(date -d '3 days ago' +%Y%m%d)"
  9. #table='alg_recsys_sample_all_test'
  10. # 处理分区配置 推荐数据间隔一天生产,所以5日0点使用3日0-23点数据生产new模型数据
  11. begin_early_2_Str="$(date -d '2 days ago' +%Y%m%d)"
  12. end_early_2_Str="$(date -d '2 days ago' +%Y%m%d)"
  13. beginHhStr=00
  14. endHhStr=23
  15. max_hour=05
  16. max_minute=00
  17. # 各节点产出hdfs文件绝对路径
  18. # 源数据文件
  19. originDataPath=/dw/recommend/model/13_sample_data/
  20. # 特征值
  21. valueDataPath=/dw/recommend/model/14_feature_data/
  22. # 特征分桶
  23. bucketDataPath=/dw/recommend/model/16_train_data/
  24. # 模型数据路径
  25. MODEL_PATH=/root/joe/recommend-emr-dataprocess/model
  26. # 预测路径
  27. PREDICT_PATH=/root/joe/recommend-emr-dataprocess/predict
  28. # 历史线上正在使用的模型数据路径
  29. LAST_MODEL_HOME=/root/joe/model_online
  30. # 模型数据文件前缀
  31. model_name=akaqjl8
  32. # fm模型
  33. FM_HOME=/root/sunmingze/alphaFM/bin
  34. # hadoop
  35. HADOOP=/opt/apps/HADOOP-COMMON/hadoop-common-current/bin/hadoop
  36. # 0 对比AUC 前置对比2日模型数据 与 线上模型数据效果对比,如果2日模型优于线上,更新线上模型
  37. #echo "$(date +%Y-%m-%d_%H-%M-%S)----------step0------------开始对比,新:${MODEL_PATH}/${model_name}_${today_early_3}.txt,与线上online模型数据auc效果"
  38. #$HADOOP fs -text ${bucketDataPath}/${begin_early_2_Str}/* | ${FM_HOME}/bin/fm_predict -m ${LAST_MODEL_HOME}/model_online.txt -dim 8 -core 8 -out ${PREDICT_PATH}/${model_name}_${today}_online.txt
  39. #$HADOOP fs -text ${bucketDataPath}/${begin_early_2_Str}/* | ${FM_HOME}/bin/fm_predict -m ${MODEL_PATH}/${model_name}_${today_early_3}.txt -dim 8 -core 8 -out ${PREDICT_PATH}/${model_name}_${today}_new.txt
  40. #
  41. #online_auc=`cat ${PREDICT_PATH}/${model_name}_${today}_online.txt | /root/sunmingze/AUC/AUC`
  42. #if [ $? -ne 0 ]; then
  43. # echo "推荐线上模型AUC计算失败"
  44. ## /root/anaconda3/bin/python ad/ad_monitor_util.py "线上模型AUC计算失败"
  45. # exit 1
  46. #fi
  47. #
  48. #new_auc=`${PREDICT_PATH}/${model_name}_${today}_new.txt | /root/sunmingze/AUC/AUC`
  49. #if [ $? -ne 0 ]; then
  50. # echo "推荐新模型AUC计算失败"
  51. ## /root/anaconda3/bin/python ad/ad_monitor_util.py "新模型AUC计算失败"
  52. # exit 1
  53. #fi
  54. #
  55. #
  56. ## 1 对比auc数据判断是否更新线上模型
  57. #if [ "$online_auc" \< "$new_auc" ]; then
  58. # echo "新模型优于线上模型: 线上模型AUC: ${online_auc}, 新模型AUC: ${new_auc}"
  59. # # todo 模型格式转换
  60. #
  61. # # todo 模型文件上传OSS
  62. #
  63. # # todo 本地保存最新的线上使用的模型,用于下一次的AUC验证
  64. ## /root/anaconda3/bin/python ad/ad_monitor_util.py "新模型优于线上模型: 线上模型AUC: ${online_auc}, 新模型AUC: ${new_auc}"
  65. #else
  66. # echo "新模型不如线上模型: 线上模型AUC: ${online_auc}, 新模型AUC: ${new_auc}"
  67. ## /root/anaconda3/bin/python ad/ad_monitor_util.py "新模型不如线上模型: 线上模型AUC: ${online_auc}, 新模型AUC: ${new_auc}"
  68. #fi
  69. # 2 判断上游表是否生产完成,最长等待到12点
  70. # shellcheck disable=SC2154
  71. echo "$(date +%Y-%m-%d_%H-%M-%S)----------step1------------开始校验是否生产完数据,分区信息:beginStr:${begin_early_2_Str}${beginHhStr},endStr:${end_early_2_Str}${endHhStr}"
  72. while true; do
  73. python_return_code=$(python /root/joe/recommend-emr-dataprocess/qiaojialiang/checkHiveDataUtil.py --table ${table} --beginStr ${begin_early_2_Str}${beginHhStr} --endStr ${end_early_2_Str}${endHhStr})
  74. echo "python 返回值:${python_return_code}"
  75. if [ $python_return_code -eq 0 ]; then
  76. echo "Python程序返回0,校验存在数据,退出循环。"
  77. break
  78. fi
  79. echo "Python程序返回非0值,不存在数据,等待五分钟后再次调用。"
  80. sleep 300
  81. current_hour=$(date +%H)
  82. current_minute=$(date +%M)
  83. # shellcheck disable=SC2039
  84. if (( current_hour > max_hour || (current_hour == max_hour && current_minute >= max_minute) )); then
  85. echo "最长等待时间已到,失败:${current_hour}-${current_minute}"
  86. exit 1
  87. fi
  88. done
  89. # 3 生产原始数据
  90. echo "$(date +%Y-%m-%d_%H-%M-%S)----------step2------------开始根据${table}生产原始数据"
  91. /opt/apps/SPARK2/spark-2.4.8-hadoop3.2-1.0.8/bin/spark-class2 org.apache.spark.deploy.SparkSubmit \
  92. --class com.aliyun.odps.spark.examples.makedata_qiao.makedata_13_originData_20240705 \
  93. --master yarn --driver-memory 1G --executor-memory 2G --executor-cores 1 --num-executors 16 \
  94. ../target/spark-examples-1.0.0-SNAPSHOT-shaded.jar \
  95. tablePart:64 repartition:32 \
  96. beginStr:${begin_early_2_Str}${beginHhStr} endStr:${end_early_2_Str}${endHhStr} \
  97. savePath:${originDataPath} \
  98. table:${table}
  99. if [ $? -ne 0 ]; then
  100. echo "Spark原始样本生产任务执行失败"
  101. exit 1
  102. else
  103. echo "spark原始样本生产执行成功"
  104. fi
  105. # 4 特征值拼接
  106. echo "$(date +%Y-%m-%d_%H-%M-%S)----------step3------------开始特征值拼接"
  107. /opt/apps/SPARK2/spark-2.4.8-hadoop3.2-1.0.8/bin/spark-class2 org.apache.spark.deploy.SparkSubmit \
  108. --class com.aliyun.odps.spark.examples.makedata_qiao.makedata_14_valueData_20240705 \
  109. --master yarn --driver-memory 1G --executor-memory 3G --executor-cores 1 --num-executors 32 \
  110. ../target/spark-examples-1.0.0-SNAPSHOT-shaded.jar \
  111. readPath:${originDataPath} \
  112. savePath:${valueDataPath} \
  113. beginStr:${begin_early_2_Str} endStr:${end_early_2_Str} repartition:1000
  114. if [ $? -ne 0 ]; then
  115. echo "Spark特征值拼接处理任务执行失败"
  116. exit 1
  117. else
  118. echo "spark特征值拼接处理执行成功"
  119. fi
  120. # 5 特征分桶
  121. echo "$(date +%Y-%m-%d_%H-%M-%S)----------step4------------根据特征分桶生产重打分特征数据"
  122. /opt/apps/SPARK2/spark-2.4.8-hadoop3.2-1.0.8/bin/spark-class2 org.apache.spark.deploy.SparkSubmit \
  123. --class com.aliyun.odps.spark.examples.makedata_qiao.makedata_16_bucketData_20240705 \
  124. --master yarn --driver-memory 2G --executor-memory 4G --executor-cores 1 --num-executors 16 \
  125. ../target/spark-examples-1.0.0-SNAPSHOT-shaded.jar \
  126. readPath:${valueDataPath} \
  127. savePath:${bucketDataPath} \
  128. beginStr:${begin_early_2_Str} endStr:${end_early_2_Str} repartition:1000
  129. if [ $? -ne 0 ]; then
  130. echo "Spark特征分桶处理任务执行失败"
  131. exit 1
  132. else
  133. echo "spark特征分桶处理执行成功"
  134. fi
  135. # 6 模型训练
  136. echo "$(date +%Y-%m-%d_%H-%M-%S)----------step5------------开始模型训练"
  137. $HADOOP fs -text ${bucketDataPath}/${begin_early_2_Str}/* | ${FM_HOME}/fm_train -m ${MODEL_PATH}/${model_name}_${begin_early_2_Str}.txt -dim 1,1,8 -im ${LAST_MODEL_HOME}/model_online.txt -core 8
  138. if [ $? -ne 0 ]; then
  139. echo "模型训练失败"
  140. # /root/anaconda3/bin/python ad/ad_monitor_util.py "推荐模型训练失败"
  141. exit 1
  142. fi
  143. echo "$(date +%Y-%m-%d_%H-%M-%S)----------step6------------模型训练完成:${MODEL_PATH}/${model_name}_${begin_early_2_Str}.txt"