Skip to content

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

模型上线

parent e4a6d718
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
shell/service_train.sh
View file @ b8da2eaf
cd /data/files/wideAndDeep/tensorServing_models && rm -rf service_copy && mv service service_copy && mkdir service cd /data/files/wideAndDeep/tensorServing_models && rm -rf service_copy && mv service service_copy && mkdir service
source /srv/envs/serviceRec/bin/activate source /srv/envs/serviceRec/bin/activate
python /srv/apps/serviceRec/train/train_service_sk_tf2_distibute.py > /srv/apps/serviceRec/logs/train_service_log.log python /srv/apps/serviceRec/train/train_service_sk_tf2_distibute.py >/srv/apps/serviceRec/logs/train_service_log.log 2>&1
\ No newline at end of file \ No newline at end of file
train/train_service_sk_tf2_distibute.py
View file @ b8da2eaf
...@@ -259,7 +259,7 @@ eval_spec = tf.estimator.EvalSpec(input_fn = lambda: input_fn(DATA_DIR + 'eval_s ...@@ -259,7 +259,7 @@ eval_spec = tf.estimator.EvalSpec(input_fn = lambda: input_fn(DATA_DIR + 'eval_s
# wideAndDeepModel = tf.contrib.estimator.add_metrics(wideAndDeepModel, my_auc) # wideAndDeepModel = tf.contrib.estimator.add_metrics(wideAndDeepModel, my_auc)
tf.estimator.train_and_evaluate(wideAndDeepModel, train_spec, eval_spec) # tf.estimator.train_and_evaluate(wideAndDeepModel, train_spec, eval_spec)
wideAndDeepModel.evaluate(lambda: input_fn(DATA_DIR + 'eval_samples.csv', 1, False, 2 ** 15)) wideAndDeepModel.evaluate(lambda: input_fn(DATA_DIR + 'eval_samples.csv', 1, False, 2 ** 15))
......
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