Skip to content

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

模型上线

parent 3a0574b3
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
train/train_service_sk_tf2_distibute.py
View file @ da5dfa7c
...@@ -10,8 +10,8 @@ start = time.time() ...@@ -10,8 +10,8 @@ start = time.time()
DATA_DIR = '/data/files/wideAndDeep/trainData/' DATA_DIR = '/data/files/wideAndDeep/trainData/'
CHECKPOINTS_DIR = '/data/files/wideAndDeep/checkpoints/' CHECKPOINTS_DIR = '/data/files/wideAndDeep/checkpoints/'
# if os.path.exists(CHECKPOINTS_DIR): if os.path.exists(CHECKPOINTS_DIR):
# os.rmdir(CHECKPOINTS_DIR) shutil.rmtree(CHECKPOINTS_DIR)
MODEL_BASE_DIR = '/data/files/wideAndDeep/tensorServing_models/service' MODEL_BASE_DIR = '/data/files/wideAndDeep/tensorServing_models/service'
...@@ -238,7 +238,7 @@ early_stopping = tf.estimator.experimental.stop_if_no_increase_hook(wideAndDeepM ...@@ -238,7 +238,7 @@ early_stopping = tf.estimator.experimental.stop_if_no_increase_hook(wideAndDeepM
hooks = [early_stopping] hooks = [early_stopping]
train_spec = tf.estimator.TrainSpec(input_fn = lambda: input_fn(DATA_DIR + 'train_samples.csv', 1, True, 2048), hooks = hooks) train_spec = tf.estimator.TrainSpec(input_fn = lambda: input_fn(DATA_DIR + 'train_samples.csv', 1, False, 2048), hooks = hooks)
serving_feature_spec = tf.feature_column.make_parse_example_spec( serving_feature_spec = tf.feature_column.make_parse_example_spec(
linear_feature_columns + dnn_feature_columns) linear_feature_columns + dnn_feature_columns)
......
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