Only training needs shuffle (#1277)

* only training needs shuffle * more fix
sql-machine-learning · Oct 9, 2019 · c1b95d0 · c1b95d0
1 parent fd8b268
commit c1b95d0
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 8 deletions.
diff --git a/model_zoo/cifar10_functional_api/cifar10_functional_api.py b/model_zoo/cifar10_functional_api/cifar10_functional_api.py
@@ -136,7 +136,7 @@ def _parse_data(record):
 
     dataset = dataset.map(_parse_data)
 
-    if mode != Mode.PREDICTION:
+    if mode == Mode.TRAINING:
         dataset = dataset.shuffle(buffer_size=1024)
     return dataset
 

diff --git a/model_zoo/cifar10_subclass/cifar10_subclass.py b/model_zoo/cifar10_subclass/cifar10_subclass.py
@@ -156,7 +156,7 @@ def _parse_data(record):
 
     dataset = dataset.map(_parse_data)
 
-    if mode != Mode.PREDICTION:
+    if mode == Mode.TRAINING:
         dataset = dataset.shuffle(buffer_size=1024)
     return dataset
 

diff --git a/model_zoo/deepfm_edl_embedding/deepfm_edl_embedding.py b/model_zoo/deepfm_edl_embedding/deepfm_edl_embedding.py
@@ -94,7 +94,7 @@ def _parse_data(record):
 
     dataset = dataset.map(_parse_data)
 
-    if mode != Mode.PREDICTION:
+    if mode == Mode.TRAINING:
         dataset = dataset.shuffle(buffer_size=1024)
     return dataset
 

diff --git a/model_zoo/deepfm_functional_api/deepfm_functional_api.py b/model_zoo/deepfm_functional_api/deepfm_functional_api.py
@@ -105,7 +105,7 @@ def _parse_data(record):
 
     dataset = dataset.map(_parse_data)
 
-    if mode != Mode.PREDICTION:
+    if mode == Mode.TRAINING:
         dataset = dataset.shuffle(buffer_size=1024)
     return dataset
 

diff --git a/model_zoo/mnist_functional_api/mnist_functional_api.py b/model_zoo/mnist_functional_api/mnist_functional_api.py
@@ -76,7 +76,7 @@ def _parse_data(record):
 
     dataset = dataset.map(_parse_data)
 
-    if mode != Mode.PREDICTION:
+    if mode == Mode.TRAINING:
         dataset = dataset.shuffle(buffer_size=1024)
     return dataset
 

diff --git a/model_zoo/mnist_subclass/mnist_subclass.py b/model_zoo/mnist_subclass/mnist_subclass.py
@@ -70,7 +70,7 @@ def _parse_data(record):
 
     dataset = dataset.map(_parse_data)
 
-    if mode != Mode.PREDICTION:
+    if mode == Mode.TRAINING:
         dataset = dataset.shuffle(buffer_size=1024)
     return dataset
 

diff --git a/model_zoo/odps_iris_dnn_model/odps_iris_dnn_model.py b/model_zoo/odps_iris_dnn_model/odps_iris_dnn_model.py
@@ -58,7 +58,7 @@ def _get_features_without_labels(
 
     dataset = dataset.map(_parse_data)
 
-    if mode != Mode.PREDICTION:
+    if mode == Mode.TRAINING:
         dataset = dataset.shuffle(buffer_size=200)
     return dataset
 

diff --git a/model_zoo/resnet50_subclass/resnet50_subclass.py b/model_zoo/resnet50_subclass/resnet50_subclass.py
@@ -201,7 +201,7 @@ def _parse_data(record):
 
     dataset = dataset.map(_parse_data)
 
-    if mode != Mode.PREDICTION:
+    if mode == Mode.TRAINING:
         dataset = dataset.shuffle(buffer_size=1024)
     return dataset