change test file

tensnsorflow/multi_hot.py

@@ -326,8 +326,7 @@ def get_predict(date,value_map,app_list_map,leve2_map,leve3_map):
 
 
     rdd.persist(storageLevel= StorageLevel.MEMORY_ONLY_SER)
-    print("esmm pre count")
-    print(rdd.count())
+
     # native_pre = spark.createDataFrame(rdd.filter(lambda x:x[0] == 0).map(lambda x:(x[3],x[4],x[5],x[17])))\
     #     .toDF("city","uid","cid_id","number")
     # print("native csv")

tensnsorflow/train_tag.py

@@ -133,8 +133,8 @@ def model_fn(features, labels, mode, params):
     tag5_list = features['tag5_list']
     tag6_list = features['tag6_list']
     tag7_list = features['tag7_list']
-    search_tag2_list = features['search_tag2']
-    search_tag3_list = features['search_tag3']
+    search_tag2_list = features['search_tag2_list']
+    search_tag3_list = features['search_tag3_list']
 
     if FLAGS.task_type != "infer":
         y = labels['y']