Skip to content

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

change test file

parent 4e518ec4
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
tensnsorflow/train_tag.py
View file @ 862bbbfb
...@@ -64,8 +64,8 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False): ...@@ -64,8 +64,8 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False):
"tag5_list": tf.VarLenFeature(tf.int64), "tag5_list": tf.VarLenFeature(tf.int64),
"tag6_list": tf.VarLenFeature(tf.int64), "tag6_list": tf.VarLenFeature(tf.int64),
"tag7_list": tf.VarLenFeature(tf.int64), "tag7_list": tf.VarLenFeature(tf.int64),
"search_tag2_list": tf.VarLenFeature(tf.int64), "search_tag2": tf.VarLenFeature(tf.int64),
"search_tag3_list": tf.VarLenFeature(tf.int64) "search_tag3": 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')
...@@ -133,8 +133,8 @@ def model_fn(features, labels, mode, params): ...@@ -133,8 +133,8 @@ def model_fn(features, labels, mode, params):
tag5_list = features['tag5_list'] tag5_list = features['tag5_list']
tag6_list = features['tag6_list'] tag6_list = features['tag6_list']
tag7_list = features['tag7_list'] tag7_list = features['tag7_list']
search_tag2_list = features['search_tag2_list'] search_tag2_list = features['search_tag2']
search_tag3_list = features['search_tag3_list'] search_tag3_list = features['search_tag3']
if FLAGS.task_type != "infer": if FLAGS.task_type != "infer":
y = labels['y'] y = labels['y']
......
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