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

fix path bug and add some optimization (#3749)

上级 c656be51
#! /bin/bash
set -e
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
gzip -d reviews_Electronics_5.json.gz
wget -c http://snap.stanford.edu/data/amazon/productGraph/categoryFiles/meta_Electronics.json.gz
gzip -d meta_Electronics.json.gz
echo "download data successful"
echo "download data successfully"
cd ..
python convert_pd.py
......
......@@ -136,7 +136,7 @@ def train():
if (global_step > 400000 and global_step % PRINT_STEP == 0) or (
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))
feed_var_name = [
"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.
先完成此消息的编辑!
想要评论请 注册