模型上线

train/train_service_sk_tf2_distibute.py

@@ -89,11 +89,10 @@ def input_fn(csv_path, epoch, shuffle, batch_size):
 
     dataset = dataset.map(parse_line, num_parallel_calls = 8).cache()
     dataset = dataset.padded_batch(batch_size, padded_shapes, padding_values=padding_values)
-    dataset = dataset.prefetch(512 * 100).repeat(epoch)
     if shuffle:
-        dataset.shuffle(2048)
+        dataset = dataset.shuffle(2048)
 
-    return dataset
+    return dataset.prefetch(512 * 100).repeat(epoch)
 
 boundaries = [0, 10, 100]
 ITEM_NUMERIC_click_count_sum_fc = tf.feature_column.bucketized_column(tf.feature_column.numeric_column('ITEM_NUMERIC_click_count_sum'), boundaries)