Skip to content

F
ffm-baseline
Commit 1bcb9ba2 authored by Your Name's avatar Your Name
Browse files
Options

test dist predict

parent 4bec895f
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
eda/esmm/Model_pipline/dist_predict.py
View file @ 1bcb9ba2
...@@ -87,8 +87,10 @@ def model_fn(features, labels, mode, params): ...@@ -87,8 +87,10 @@ def model_fn(features, labels, mode, params):
tag6_list = features['tag6_list'] tag6_list = features['tag6_list']
tag7_list = features['tag7_list'] tag7_list = features['tag7_list']
number = features['number'] number = features['number']
uid = str(features['uid']) uid2 = features['uid']
city = str(features['city']) uid = tf.cast(uid2, dtype=tf.string)
city2 = features['city']
city = tf.cast(city2, dtype=tf.string)
cid_id = features['cid_id'] cid_id = features['cid_id']
......
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