Skip to content

F
ffm-baseline
Commit 7331edb3 authored by 张彦钊's avatar 张彦钊
Browse files
Options

change test file

parent ec4e2fd3
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
tensnsorflow/train_multi.py
View file @ 7331edb3
...@@ -50,8 +50,7 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False): ...@@ -50,8 +50,7 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False):
"y": tf.FixedLenFeature([], tf.float32), "y": tf.FixedLenFeature([], tf.float32),
"z": tf.FixedLenFeature([], tf.float32), "z": tf.FixedLenFeature([], tf.float32),
"ids": tf.FixedLenFeature([FLAGS.field_size], tf.int64), "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) parsed = tf.parse_single_example(record, features)
y = parsed.pop('y') y = parsed.pop('y')
...@@ -292,8 +291,8 @@ def main(_): ...@@ -292,8 +291,8 @@ def main(_):
#------init Envs------ #------init Envs------
path = "hdfs:///strategy/esmm/" path = "hdfs:///strategy/esmm/"
tr_files = ["hdfs:///strategy/va.tfrecord"] # tr_files = ["hdfs:///strategy/va.tfrecord"]
# tr_files = [path+"tr/part-r-00000"] tr_files = [path+"tr/part-r-00000"]
va_files = [path+"va/part-r-00000"] va_files = [path+"va/part-r-00000"]
te_files = ["%s/part-r-00000" % FLAGS.hdfs_dir] te_files = ["%s/part-r-00000" % FLAGS.hdfs_dir]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment