submit.sh 2.59 KB
Newer Older
高雅喆's avatar
高雅喆 committed
1
#! /bin/bash
2
git checkout master
张彦钊's avatar
张彦钊 committed
3
PYTHON_PATH=/opt/anaconda3/envs/esmm/bin/python
张彦钊's avatar
张彦钊 committed
4
MODEL_PATH=/srv/apps/ffm-baseline/eda/esmm/Model_pipline
张彦钊's avatar
张彦钊 committed
5
DATA_PATH=/home/gmuser/esmm
高雅喆's avatar
高雅喆 committed
6 7

echo "rm leave tfrecord"
高雅喆's avatar
高雅喆 committed
8 9 10 11
rm ${DATA_PATH}/tr/*
rm ${DATA_PATH}/va/*
rm ${DATA_PATH}/native/*
rm ${DATA_PATH}/nearby/*
12
rm -r ${DATA_PATH}/model_ckpt/DeepCvrMTL/20*
高雅喆's avatar
高雅喆 committed
13

14
echo "data"
15
${PYTHON_PATH} ${MODEL_PATH}/feature.py > ${DATA_PATH}/feature.log
高雅喆's avatar
高雅喆 committed
16 17

echo "csv to tfrecord"
18 19 20 21
${PYTHON_PATH} ${MODEL_PATH}/to_tfrecord.py --input_dir=${DATA_PATH}/tr/ --output_dir=${DATA_PATH}/tr/
${PYTHON_PATH} ${MODEL_PATH}/to_tfrecord.py --input_dir=${DATA_PATH}/va/ --output_dir=${DATA_PATH}/va/
${PYTHON_PATH} ${MODEL_PATH}/to_tfrecord.py --input_dir=${DATA_PATH}/native/ --output_dir=${DATA_PATH}/native/
${PYTHON_PATH} ${MODEL_PATH}/to_tfrecord.py --input_dir=${DATA_PATH}/nearby/ --output_dir=${DATA_PATH}/nearby/
高雅喆's avatar
高雅喆 committed
22

高雅喆's avatar
高雅喆 committed
23 24 25 26
cat ${DATA_PATH}/tr/*.tfrecord > ${DATA_PATH}/tr/tr.tfrecord
cat ${DATA_PATH}/va/*.tfrecord > ${DATA_PATH}/va/va.tfrecord
cat ${DATA_PATH}/native/*.tfrecord > ${DATA_PATH}/native/native.tfrecord
cat ${DATA_PATH}/nearby/*.tfrecord > ${DATA_PATH}/nearby/nearby.tfrecord
高雅喆's avatar
高雅喆 committed
27

高雅喆's avatar
高雅喆 committed
28 29 30 31
rm ${DATA_PATH}/tr/tr_*
rm ${DATA_PATH}/va/va_*
rm ${DATA_PATH}/native/native_*
rm ${DATA_PATH}/nearby/nearby_*
高雅喆's avatar
高雅喆 committed
32

高雅喆's avatar
高雅喆 committed
33

高雅喆's avatar
高雅喆 committed
34
echo "train..."
35
${PYTHON_PATH} ${MODEL_PATH}/train.py --ctr_task_wgt=0.5 --learning_rate=0.0001 --deep_layers=512,256,128,64,32 --dropout=0.3,0.3,0.3,0.3,0.3 --optimizer=Adam --num_epochs=1 --embedding_size=16 --batch_size=2000 --field_size=15 --feature_size=300000 --l2_reg=0.005 --log_steps=100 --num_threads=36 --model_dir=${DATA_PATH}/model_ckpt/DeepCvrMTL/ --data_dir=${DATA_PATH} --task_type=train
36

高雅喆's avatar
高雅喆 committed
37 38

echo "infer native..."
39
${PYTHON_PATH} ${MODEL_PATH}/train.py --ctr_task_wgt=0.5 --learning_rate=0.0001 --deep_layers=512,256,128,64,32 --dropout=0.3,0.3,0.3,0.3,0.3 --optimizer=Adam --num_epochs=1 --embedding_size=16 --batch_size=2000 --field_size=15 --feature_size=300000 --l2_reg=0.005 --log_steps=100 --num_threads=36 --model_dir=${DATA_PATH}/model_ckpt/DeepCvrMTL/ --data_dir=${DATA_PATH}/native --task_type=infer > ${DATA_PATH}/native_infer.log
高雅喆's avatar
高雅喆 committed
40 41 42


echo "infer nearby..."
43
${PYTHON_PATH} ${MODEL_PATH}/train.py --ctr_task_wgt=0.5 --learning_rate=0.0001 --deep_layers=512,256,128,64,32 --dropout=0.3,0.3,0.3,0.3,0.3 --optimizer=Adam --num_epochs=1 --embedding_size=16 --batch_size=2000 --field_size=15 --feature_size=300000 --l2_reg=0.005 --log_steps=100 --num_threads=36 --model_dir=${DATA_PATH}/model_ckpt/DeepCvrMTL/ --data_dir=${DATA_PATH}/nearby --task_type=infer > ${DATA_PATH}/nearby_infer.log
高雅喆's avatar
高雅喆 committed
44 45

echo "sort and 2sql"
张彦钊's avatar
张彦钊 committed
46
${PYTHON_PATH} ${MODEL_PATH}/to_database.py > ${DATA_PATH}/insert_database.log