Skip to content

F
ffm-baseline
Commit 7cf73afa authored by Your Name's avatar Your Name
Browse files
Options

test dist predict

parent 3053abcc
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
eda/esmm/Model_pipline/dist_predict.py
View file @ 7cf73afa
......@@ -31,7 +31,7 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False):
"number": tf.VarLenFeature(tf.int64),
"uid": tf.VarLenFeature(tf.string),
"city": tf.VarLenFeature(tf.string),
"cid_id": tf.VarLenFeature(tf.string)
"cid_id": tf.VarLenFeature(tf.int64)
}
parsed = tf.parse_single_example(record, features)
y = parsed.pop('y')
......@@ -113,7 +113,7 @@ def model_fn(features, labels, mode, params):
sample_id = tf.sparse.to_dense(number)
uid = tf.sparse.to_dense(uid,default_value="")
city = tf.sparse.to_dense(city,default_value="")
cid_id = tf.sparse.to_dense(cid_id,default_value="")
cid_id = tf.sparse.to_dense(cid_id)
with tf.name_scope("CVR_Task"):
if mode == tf.estimator.ModeKeys.TRAIN:
......
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