zhangbo 1 년 전
부모
커밋
824b569a7e
2개의 변경된 파일7개의 추가작업 그리고 3개의 파일을 삭제
  1. 2 2
      zhangbo/01_train.sh
  2. 5 1
      zhangbo/03_predict.sh

+ 2 - 2
zhangbo/01_train.sh

@@ -12,6 +12,6 @@ $HADOOP fs -text ${train_path}/dt=$day/* | /root/sunmingze/alphaFM/bin/fm_train
 # -v_l1 ${v_l1} -v_l2 ${v_l2}
 
 
-# nohup sh 01_train.sh 20240222 /dw/recommend/model/12_ros_data_v3 model_jerry >p1_train.log 2>&1 &
-# nohup sh 01_train.sh 20240222 /dw/recommend/model/11_str_data_v3 model_jerry112 1,1,2 >p1_model_jerry112.log 2>&1 &
+# nohup sh 01_train.sh 20240222 /dw/recommend/model/12_ros_data_v3_noweight model_tom_noweight 1,1,0 >p1_model_tom_noweight.log 2>&1 &
 
+# nohup sh 01_train.sh 20240222 /dw/recommend/model/11_str_data_v3 model_jerry112 1,1,2 >p1_model_jerry112.log 2>&1 &

+ 5 - 1
zhangbo/03_predict.sh

@@ -6,8 +6,9 @@ day=$1
 train_path=$2
 model_name=$3
 output_file=$4
+bias=$5
 HADOOP="/opt/apps/HADOOP-COMMON/hadoop-common-current/bin/hadoop"
-$HADOOP fs -text ${train_path}/dt=$day/* | /root/sunmingze/alphaFM/bin/fm_predict -m model/$model_name -dim 0 -core 8 -out predict/${output_file}_$day.txt
+$HADOOP fs -text ${train_path}/dt=$day/* | /root/sunmingze/alphaFM/bin/fm_predict -m model/$model_name -dim ${bias} -core 8 -out predict/${output_file}_$day.txt
 cat predict/${output_file}_$day.txt | /root/sunmingze/AUC/AUC
 
 # nohup sh 03_predict.sh 20240226 /dw/recommend/model/12_ros_data_v3/ model_jerry_20240225.txt model_jerry >p3_pred.log 2>&1 &
@@ -15,3 +16,6 @@ cat predict/${output_file}_$day.txt | /root/sunmingze/AUC/AUC
 # nohup sh 03_predict.sh 20240226 /dw/recommend/model/11_str_data_v3/ model_str_tom_20240222.txt model_str_tom >p3.log 2>&1 &
 
 # nohup sh 03_predict.sh 20240226 /dw/recommend/model/11_str_data_v3/ model_str_tom_20240225.txt model_str_tom >p3.123.log 2>&1 &
+
+
+# nohup sh 03_predict.sh 20240226 /dw/recommend/model/11_str_data_v3/ model_jerry112_20240222.txt model_jerry112 1,1,2 >p3_model_jerry112.log 2>&1 &