test dist predict

0dfd2cde · Your Name · 4d82b9a8 · 0dfd2cde
Commit 0dfd2cde authored Jun 19, 2019 by Your Name
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

dist_predict.py eda/esmm/Model_pipline/dist_predict.py +2 -2

No files found.
--- a/eda/esmm/Model_pipline/dist_predict.py
+++ b/eda/esmm/Model_pipline/dist_predict.py
@@ -111,8 +111,8 @@ def model_fn(features, labels, mode, params):
                              tag2, tag3, tag4, tag5, tag6, tag7], axis=1)

        sample_id = tf.sparse.to_dense(number)
-        uid = tf.sparse.to_dense(uid)
-        city = tf.sparse.to_dense(city)
+        uid = tf.sparse.to_dense(uid,default_value="")
+        city = tf.sparse.to_dense(city,default_value="")
        cid_id = tf.sparse.to_dense(cid_id)

    with tf.name_scope("CVR_Task"):