change test file

eda/esmm/Model_pipline/feature_engineering.py

@@ -76,32 +76,38 @@ def con_sql(db,sql):
     return df
 
 
-def train_yingshe(x):
-    tmp = (x[0], float(x[1]), float(x[2]), app_list_func(x[3], app_list_map), app_list_func(x[4], leve2_map),
-                   app_list_func(x[5], leve3_map), app_list_func(x[6], leve2_map), app_list_func(x[7], leve2_map),
-                   app_list_func(x[8], leve2_map), app_list_func(x[9], leve2_map), app_list_func(x[10], leve2_map),
-                   app_list_func(x[11], leve2_map), app_list_func(x[12], leve2_map),
-                   [value_map[x[0]], value_map[x[13]], value_map[x[14]], value_map[x[15]], value_map[x[16]],
-                    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]]])
+def train_yingshe(y):
+    tmp = []
+    for x in y:
+        i = (x[0], float(x[1]), float(x[2]), app_list_func(x[3], app_list_map), app_list_func(x[4], leve2_map),
+                       app_list_func(x[5], leve3_map), app_list_func(x[6], leve2_map), app_list_func(x[7], leve2_map),
+                       app_list_func(x[8], leve2_map), app_list_func(x[9], leve2_map), app_list_func(x[10], leve2_map),
+                       app_list_func(x[11], leve2_map), app_list_func(x[12], leve2_map),
+                       [value_map[x[0]], value_map[x[13]], value_map[x[14]], value_map[x[15]], value_map[x[16]],
+                        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
 
-def pre_yingshe(x):
-    tmp = (x[0], float(x[1]), float(x[2]), x[3], x[4], x[5],
-     app_list_func(x[6], app_list_map), app_list_func(x[7], leve2_map),
-     app_list_func(x[8], leve3_map), app_list_func(x[9], leve2_map),
-     app_list_func(x[10], leve2_map), app_list_func(x[11], leve2_map),
-     app_list_func(x[12], leve2_map), app_list_func(x[13], leve2_map),
-     app_list_func(x[14], leve2_map), app_list_func(x[15], leve2_map),
-     [value_map.get(validate_date, 1), value_map.get(x[16], 2),
-      value_map.get(x[17], 3), value_map.get(x[18], 4),
-      value_map.get(x[19], 5), value_map.get(x[20], 6),
-      value_map.get(x[21], 7), value_map.get(x[22], 8),
-      value_map.get(x[23], 9), value_map.get(x[24], 10),
-      value_map.get(x[25], 11), value_map.get(x[26], 12),
-      value_map.get(x[27], 13), value_map.get(x[28], 14),
-      value_map.get(x[29], 15)
-      ])
+def pre_yingshe(y):
+    tmp = []
+    for x in y:
+        i = (x[0], float(x[1]), float(x[2]), x[3], x[4], x[5],
+         app_list_func(x[6], app_list_map), app_list_func(x[7], leve2_map),
+         app_list_func(x[8], leve3_map), app_list_func(x[9], leve2_map),
+         app_list_func(x[10], leve2_map), app_list_func(x[11], leve2_map),
+         app_list_func(x[12], leve2_map), app_list_func(x[13], leve2_map),
+         app_list_func(x[14], leve2_map), app_list_func(x[15], leve2_map),
+         [value_map.get(validate_date, 1), value_map.get(x[16], 2),
+          value_map.get(x[17], 3), value_map.get(x[18], 4),
+          value_map.get(x[19], 5), value_map.get(x[20], 6),
+          value_map.get(x[21], 7), value_map.get(x[22], 8),
+          value_map.get(x[23], 9), value_map.get(x[24], 10),
+          value_map.get(x[25], 11), value_map.get(x[26], 12),
+          value_map.get(x[27], 13), value_map.get(x[28], 14),
+          value_map.get(x[29], 15)
+          ])
+        tmp.append(i)
     return tmp
 
 def feature_engineer():