未验证 提交 9445121f 编写于 作者: H hutuxian 提交者: GitHub

fix path bug and add some optimization (#3749)

上级 c656be51
#! /bin/bash #! /bin/bash
set -e
echo "begin download data" echo "begin download data"
mkdir raw_data && cd raw_data mkdir raw_data
cd raw_data
wget -c http://snap.stanford.edu/data/amazon/productGraph/categoryFiles/reviews_Electronics_5.json.gz wget -c http://snap.stanford.edu/data/amazon/productGraph/categoryFiles/reviews_Electronics_5.json.gz
gzip -d reviews_Electronics_5.json.gz gzip -d reviews_Electronics_5.json.gz
wget -c http://snap.stanford.edu/data/amazon/productGraph/categoryFiles/meta_Electronics.json.gz wget -c http://snap.stanford.edu/data/amazon/productGraph/categoryFiles/meta_Electronics.json.gz
gzip -d meta_Electronics.json.gz gzip -d meta_Electronics.json.gz
echo "download data successful" echo "download data successfully"
cd .. cd ..
python convert_pd.py python convert_pd.py
......
...@@ -136,7 +136,7 @@ def train(): ...@@ -136,7 +136,7 @@ def train():
if (global_step > 400000 and global_step % PRINT_STEP == 0) or ( if (global_step > 400000 and global_step % PRINT_STEP == 0) or (
global_step <= 400000 and global_step % 50000 == 0): global_step <= 400000 and global_step % 50000 == 0):
save_dir = os.path.join(args.model_dir, "/global_step_" + str( save_dir = os.path.join(args.model_dir, "global_step_" + str(
global_step)) global_step))
feed_var_name = [ feed_var_name = [
"hist_item_seq", "hist_cat_seq", "target_item", "hist_item_seq", "hist_cat_seq", "target_item",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册