change test file

11de18e1 · 张彦钊 · fc93711d · 11de18e1
Commit 11de18e1 authored May 30, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

feature_engineering.py eda/esmm/Model_pipline/feature_engineering.py +2 -2

No files found.
--- a/eda/esmm/Model_pipline/feature_engineering.py
+++ b/eda/esmm/Model_pipline/feature_engineering.py
@@ -87,7 +87,7 @@ def train_yingshe(y):
                        value_map[x[17]], value_map[x[18]], value_map[x[19]], value_map[x[20]], value_map[x[21]],
                        value_map[x[22]], value_map[x[23]], value_map[x[24]], value_map[x[25]], value_map[x[26]]])
        tmp.append(i)
-    return tmp
+    return iter(tmp)
 def pre_yingshe(y):
    tmp = []
@@ -108,7 +108,7 @@ def pre_yingshe(y):
          value_map.get(x[29], 15)
          ])
        tmp.append(i)
-    return tmp
+    return iter(tmp)
 def feature_engineer():
    apps_number, app_list_map, level2_number, leve2_map, level3_number, leve3_map = get_map()