|  | @@ -2,12 +2,13 @@
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  # 模型训练,预测,计算AUC脚本
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -# set -x
 | 
	
		
			
				|  |  | +set -x
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  begin_date=$1
 | 
	
		
			
				|  |  |  end_date=$2
 | 
	
		
			
				|  |  |  model_name=$3
 | 
	
		
			
				|  |  | -dim=$4
 | 
	
		
			
				|  |  | +train_dim=$4
 | 
	
		
			
				|  |  | +predict_dim=$5
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  PROJECT_HOME=/root/zhaohp/recommend-emr-dataprocess
 | 
	
		
			
				|  |  |  HADOOP=/opt/apps/HADOOP-COMMON/hadoop-common-current/bin/hadoop
 | 
	
	
		
			
				|  | @@ -28,7 +29,7 @@ predict_auc() {
 | 
	
		
			
				|  |  |      predict_end_date=$(date -d "$end_date +1 day" +%Y%m%d)
 | 
	
		
			
				|  |  |      while [ "$predict_date" != "$predict_end_date" ]; do
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -        $HADOOP fs -text ${HDFS_TRAIN_DATE_PATH}/${predict_date}/* | ${FM_PREDICT} -m ${MODEL_PATH}/${model_name}_${train_date}.txt -dim ${dim} -core 8 -out ${PREDICT_PATH}/${model_name}_${train_date}.txt
 | 
	
		
			
				|  |  | +        $HADOOP fs -text ${HDFS_TRAIN_DATE_PATH}/${predict_date}/* | ${FM_PREDICT} -m ${MODEL_PATH}/${model_name}_${train_date}.txt -dim ${predict_dim} -core 8 -out ${PREDICT_PATH}/${model_name}_${train_date}.txt
 | 
	
		
			
				|  |  |          auc=`cat ${PREDICT_PATH}/${model_name}_${train_date}.txt | /root/sunmingze/AUC/AUC`
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |          echo "模型训练日期: ${train_date}, 模型预测日期: ${predict_date}, AUC: ${auc}, 模型路径: ${MODEL_PATH}/${model_name}_${train_date}.txt"
 | 
	
	
		
			
				|  | @@ -48,10 +49,10 @@ main() {
 | 
	
		
			
				|  |  |          echo "==================== 开始训练 $train_date 模型 ===================="
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |          if [ "$train_date" == "$begin_date" ]; then
 | 
	
		
			
				|  |  | -            $HADOOP fs -text ${HDFS_TRAIN_DATE_PATH}/${train_date}/* | ${FM_TRAIN} -m ${MODEL_PATH}/${model_name}_${train_date}.txt -dim ${dim} -core 8
 | 
	
		
			
				|  |  | +            $HADOOP fs -text ${HDFS_TRAIN_DATE_PATH}/${train_date}/* | ${FM_TRAIN} -m ${MODEL_PATH}/${model_name}_${train_date}.txt -dim ${train_dim} -core 8
 | 
	
		
			
				|  |  |          else
 | 
	
		
			
				|  |  |              yesterday=$(date -d "$train_date -1 day" +%Y%m%d)
 | 
	
		
			
				|  |  | -            $HADOOP fs -text ${HDFS_TRAIN_DATE_PATH}/${train_date}/* | ${FM_TRAIN} -m ${MODEL_PATH}/${model_name}_${train_date}.txt -dim ${dim} -core 8 -im ${MODEL_PATH}/${model_name}_${yesterday}.txt
 | 
	
		
			
				|  |  | +            $HADOOP fs -text ${HDFS_TRAIN_DATE_PATH}/${train_date}/* | ${FM_TRAIN} -m ${MODEL_PATH}/${model_name}_${train_date}.txt -dim ${train_dim} -core 8 -im ${MODEL_PATH}/${model_name}_${yesterday}.txt
 | 
	
		
			
				|  |  |          fi
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |          predict_auc
 |