fix typo

shenweichen · Jul 17, 2019 · 3aed781 · 3aed781
1 parent efdccaa
commit 3aed781
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/README.md b/README.md
@@ -8,13 +8,9 @@ In Proceedings of 28th International Joint Conference on Artificial Intelligence
 
 ----------------
 ## Operating environment
-- python==3.6
-- tensorflow-gpu==1.4.0
-- deepctr==0.4.1
-- numpy==1.15.1
-- pandas==0.22.0
-- scikit-learn==0.19.2
-- tqdm==4.19.5
+please use 
+`pip install -r requirements.txt`
+to setup the operating environment in `python3.6`.
 
 --------------------------
 ## Download dataset and preprocess

diff --git a/code/2_gen_dsin_input.py b/code/2_gen_dsin_input.py
@@ -149,7 +149,7 @@ def gen_sess_feature_dsin(row):
         sess_name = 'sess_' + str(i)
         for feat in sess_feature:
             sess_input.append(pad_sequences(
-                sess_input_dict[sess_name][feat], maxlen=SESS_COUNT, padding='post'))
+                sess_input_dict[sess_name][feat], maxlen=DSIN_SESS_MAX_LEN, padding='post'))
         sess_input_length.append(sess_input_length_dict[sess_name])
 
     model_input = [data[feat.name].values for feat in sparse_feature_list] + \

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,8 @@
+deepctr==0.4.1
+joblib==0.13.2
+numpy==1.16.4
+pandas==0.22.0
+requests==2.22.0
+scikit-learn==0.19.2
+tensorflow-gpu==1.4.0
+tqdm==4.32.2