change test file

tensnsorflow/train_multi.py

@@ -50,8 +50,7 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False):
             "y": tf.FixedLenFeature([], tf.float32),
             "z": tf.FixedLenFeature([], tf.float32),
             "ids": tf.FixedLenFeature([FLAGS.field_size], tf.int64),
-            "level2_list": tf.VarLenFeature(tf.int64)
-            # "level2_ids": tf.VarLenFeature(tf.int64)
+            "level2_ids": tf.VarLenFeature(tf.int64)
         }
         parsed = tf.parse_single_example(record, features)
         y = parsed.pop('y')
@@ -292,8 +291,8 @@ def main(_):
     #------init Envs------
     path = "hdfs:///strategy/esmm/"
 
-    tr_files = ["hdfs:///strategy/va.tfrecord"]
-    # tr_files = [path+"tr/part-r-00000"]
+    # tr_files = ["hdfs:///strategy/va.tfrecord"]
+    tr_files = [path+"tr/part-r-00000"]
     va_files = [path+"va/part-r-00000"]
     te_files = ["%s/part-r-00000" % FLAGS.hdfs_dir]