nor_train.sh 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. #!/bin/sh
  2. set -x
  3. export HADOOP_CONF_DIR=/etc/taihao-apps/hadoop-conf
  4. export JAVA_HOME=/usr/lib/jvm/java-1.8.0
  5. # params
  6. FEATURE_FILE=20250303_recsys_nor_name.txt
  7. BASE_TRAIN_DATA_PATH=/dw/recommend/model/82_recsys_nor_train_data
  8. PREDICT_RESULT_PATH=/dw/recommend/model/82_recsys_nor_predict_data
  9. MODEL_SAVE_PATH=/dw/recommend/model/82_recsys_nor_model/model_xgb
  10. start_date=20250221
  11. end_date=20250228
  12. train_data_path=""
  13. for((i=0; i<=21; i++))
  14. do
  15. data_date=$(date -d "$start_date $i day" +"%Y%m%d")
  16. if [ "$data_date" -le "$end_date" ]
  17. then
  18. one_day_data_path="${BASE_TRAIN_DATA_PATH}/${data_date}"
  19. if [[ -z $train_data_path ]]
  20. then
  21. train_data_path=$one_day_data_path
  22. else
  23. train_data_path="$train_data_path,$one_day_data_path"
  24. fi
  25. fi
  26. done
  27. ## ******* train *******
  28. /opt/apps/SPARK3/spark-3.3.1-hadoop3.2-1.0.5/bin/spark-class org.apache.spark.deploy.SparkSubmit \
  29. --class com.tzld.piaoquan.recommend.model.train_recsys_61_xgb_nor_20241209 \
  30. --master yarn --driver-memory 4G --executor-memory 10G --executor-cores 1 --num-executors 32 \
  31. --conf spark.yarn.executor.memoryoverhead=2048 \
  32. --conf spark.shuffle.service.enabled=true \
  33. --conf spark.shuffle.service.port=7337 \
  34. --conf spark.shuffle.consolidateFiles=true \
  35. --conf spark.shuffle.manager=sort \
  36. --conf spark.storage.memoryFraction=0.4 \
  37. --conf spark.shuffle.memoryFraction=0.5 \
  38. --conf spark.default.parallelism=200 \
  39. --conf spark.sql.debug.maxToStringFields=100 \
  40. /mnt/disk1/jch/recommend-model/recommend-model-produce/target/recommend-model-produce-jar-with-dependencies.jar \
  41. featureFile:${FEATURE_FILE} \
  42. trainPath:${train_data_path} \
  43. savePath:${PREDICT_RESULT_PATH} \
  44. modelPath:${MODEL_SAVE_PATH} \
  45. labelLogType:0 \
  46. labelLogBase:1.5 \
  47. eta:0.06 gamma:0.0 max_depth:5 num_round:1000 num_worker:32 repartition:20