change test file

8cb7057b · 张彦钊 · a2c7a10e · 8cb7057b
Commit 8cb7057b authored May 23, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

train_multi.py tensnsorflow/train_multi.py +2 -2

No files found.
--- a/tensnsorflow/train_multi.py
+++ b/tensnsorflow/train_multi.py
@@ -47,8 +47,8 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False):
    print('Parsing', filenames)
    def _parse_fn(record):
        features = {
-            "y": tf.FixedLenFeature([], tf.float32),
-            "z": tf.FixedLenFeature([], tf.float32),
+            "y": tf.FixedLenFeature([], tf.int64),
+            "z": tf.FixedLenFeature([], tf.int64),
            "ids": tf.FixedLenFeature([FLAGS.field_size], tf.int64),
            "level2_ids": tf.VarLenFeature(tf.int64)
        }