Commit 340d8f5b authored by 赵威's avatar 赵威

print model info

parent 8181ff5f
...@@ -29,6 +29,10 @@ def esmm_model_fn(features, labels, mode, params): ...@@ -29,6 +29,10 @@ def esmm_model_fn(features, labels, mode, params):
ctr_preds = tf.sigmoid(ctr_logits) ctr_preds = tf.sigmoid(ctr_logits)
cvr_preds = tf.sigmoid(cvr_logits) cvr_preds = tf.sigmoid(cvr_logits)
ctcvr_preds = tf.multiply(ctr_preds, cvr_preds) ctcvr_preds = tf.multiply(ctr_preds, cvr_preds)
print("@@@@@@@@@@@")
print(ctr_logits)
print(ctr_preds)
print("@@@@@@@@@@@")
# optimizer = tf.compat.v1.train.AdamOptimizer() # optimizer = tf.compat.v1.train.AdamOptimizer()
# click_label = features["click_label"] # click_label = features["click_label"]
......
...@@ -77,7 +77,8 @@ def main(): ...@@ -77,7 +77,8 @@ def main():
model_export_path = str(Path("/data/files/models/tractate/").expanduser()) model_export_path = str(Path("/data/files/models/tractate/").expanduser())
save_path = model_export(model, all_features, model_export_path) save_path = model_export(model, all_features, model_export_path)
print("save to: " + save_path) print("save to: " + save_path)
set_essm_model_save_path("tractate", save_path) # TODO save model
# set_essm_model_save_path("tractate", save_path)
record_esmm_auc_to_db("tractate", ctr_auc, ctcvr_auc, total_time, save_path) record_esmm_auc_to_db("tractate", ctr_auc, ctcvr_auc, total_time, save_path)
print("============================================================") print("============================================================")
......
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