Skip to content

S
serviceRec
Commit a2a2801e authored by 宋柯's avatar 宋柯
Browse files
Options

模型调试

parent 11fb57e9
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
train/train_service_sk.py
View file @ a2a2801e
import tensorflow as tf import tensorflow as tf
tf.logging.set_verbosity(tf.logging.INFO) tf.logging.set_verbosity(tf.logging.INFO)
import sys import sys
import os
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
BASE_DIR = '/data/files/wideAndDeep/' BASE_DIR = '/data/files/wideAndDeep/'
...@@ -79,9 +81,9 @@ def input_fn(csv_path, epoch, shuffle, batch_size): ...@@ -79,9 +81,9 @@ def input_fn(csv_path, epoch, shuffle, batch_size):
dataset = dataset.map(parse_line, num_parallel_calls=8) dataset = dataset.map(parse_line, num_parallel_calls=8)
dataset = dataset.padded_batch(batch_size, padded_shapes, padding_values=padding_values) dataset = dataset.padded_batch(batch_size, padded_shapes, padding_values=padding_values)
if shuffle: if shuffle:
dataset = dataset.shuffle(1000).prefetch(512 * 10).repeat(epoch) dataset = dataset.shuffle(1000).prefetch(512 * 100).repeat(epoch)
else: else:
dataset = dataset.prefetch(512 * 10).repeat(epoch) dataset = dataset.prefetch(512 * 100).repeat(epoch)
return dataset return dataset
......
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