change test file

9b554ea0 · 张彦钊 · 56803ce4 · 9b554ea0
Commit 9b554ea0 authored May 22, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

train_multi.py tensnsorflow/train_multi.py +3 -3

No files found.
--- a/tensnsorflow/train_multi.py
+++ b/tensnsorflow/train_multi.py
@@ -47,9 +47,9 @@ 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),
-            "ids": tf.FixedLenFeature([FLAGS.field_size], tf.int64),
+            "y": tf.VarLenFeature(tf.int64),
+            "z": tf.VarLenFeature(tf.int64),
+            "ids": tf.VarLenFeature(tf.int64),
            "level2_ids": tf.VarLenFeature(tf.int64)
        }