Skip to content

S
strategy_embedding
Commit 03290173 authored by 赵威's avatar 赵威
Browse files
Options

update inputs

parent 445b2c54
Hide whitespace changes
Inline Side-by-side
Showing with 105 additions and 105 deletions
dssm/dssm_tractate_model.py
View file @ 03290173
...@@ -278,108 +278,108 @@ def tractate_dssm_model(): ...@@ -278,108 +278,108 @@ def tractate_dssm_model():
device_sp2 = layers.Input(shape=(1, ), name="device_sp2") device_sp2 = layers.Input(shape=(1, ), name="device_sp2")
device_p2 = layers.Input(shape=(1, ), name="device_p2") device_p2 = layers.Input(shape=(1, ), name="device_p2")
card_id = layers.Input(shape=(1, ), name="card_id"), card_id = layers.Input(shape=(1, ), name="card_id")
is_pure_author = layers.Input(shape=(1, ), name="is_pure_author"), is_pure_author = layers.Input(shape=(1, ), name="is_pure_author")
is_have_pure_reply = layers.Input(shape=(1, ), name="is_have_pure_reply"), is_have_pure_reply = layers.Input(shape=(1, ), name="is_have_pure_reply")
is_have_reply = layers.Input(shape=(1, ), name="is_have_reply"), is_have_reply = layers.Input(shape=(1, ), name="is_have_reply")
content_level = layers.Input(shape=(1, ), name="content_level"), content_level = layers.Input(shape=(1, ), name="content_level")
topic_seven_click_num = layers.Input(shape=(1, ), name="topic_seven_click_num"), topic_seven_click_num = layers.Input(shape=(1, ), name="topic_seven_click_num")
topic_thirty_click_num = layers.Input(shape=(1, ), name="topic_thirty_click_num"), topic_thirty_click_num = layers.Input(shape=(1, ), name="topic_thirty_click_num")
topic_num = layers.Input(shape=(1, ), name="topic_num"), topic_num = layers.Input(shape=(1, ), name="topic_num")
seven_transform_num = layers.Input(shape=(1, ), name="seven_transform_num"), seven_transform_num = layers.Input(shape=(1, ), name="seven_transform_num")
thirty_transform_num = layers.Input(shape=(1, ), name="thirty_transform_num"), thirty_transform_num = layers.Input(shape=(1, ), name="thirty_transform_num")
favor_num = layers.Input(shape=(1, ), name="favor_num"), favor_num = layers.Input(shape=(1, ), name="favor_num")
favor_pure_num = layers.Input(shape=(1, ), name="favor_pure_num"), favor_pure_num = layers.Input(shape=(1, ), name="favor_pure_num")
vote_num = layers.Input(shape=(1, ), name="vote_num"), vote_num = layers.Input(shape=(1, ), name="vote_num")
vote_display_num = layers.Input(shape=(1, ), name="vote_display_num"), vote_display_num = layers.Input(shape=(1, ), name="vote_display_num")
reply_num = layers.Input(shape=(1, ), name="reply_num"), reply_num = layers.Input(shape=(1, ), name="reply_num")
reply_pure_num = layers.Input(shape=(1, ), name="reply_pure_num"), reply_pure_num = layers.Input(shape=(1, ), name="reply_pure_num")
one_click_num = layers.Input(shape=(1, ), name="one_click_num"), one_click_num = layers.Input(shape=(1, ), name="one_click_num")
three_click_num = layers.Input(shape=(1, ), name="three_click_num"), three_click_num = layers.Input(shape=(1, ), name="three_click_num")
seven_click_num = layers.Input(shape=(1, ), name="seven_click_num"), seven_click_num = layers.Input(shape=(1, ), name="seven_click_num")
fifteen_click_num = layers.Input(shape=(1, ), name="fifteen_click_num"), fifteen_click_num = layers.Input(shape=(1, ), name="fifteen_click_num")
thirty_click_num = layers.Input(shape=(1, ), name="thirty_click_num"), thirty_click_num = layers.Input(shape=(1, ), name="thirty_click_num")
sixty_click_num = layers.Input(shape=(1, ), name="sixty_click_num"), sixty_click_num = layers.Input(shape=(1, ), name="sixty_click_num")
ninety_click_num = layers.Input(shape=(1, ), name="ninety_click_num"), ninety_click_num = layers.Input(shape=(1, ), name="ninety_click_num")
history_click_num = layers.Input(shape=(1, ), name="history_click_num"), history_click_num = layers.Input(shape=(1, ), name="history_click_num")
one_precise_exposure_num = layers.Input(shape=(1, ), name="one_precise_exposure_num"), one_precise_exposure_num = layers.Input(shape=(1, ), name="one_precise_exposure_num")
three_precise_exposure_num = layers.Input(shape=(1, ), name="three_precise_exposure_num"), three_precise_exposure_num = layers.Input(shape=(1, ), name="three_precise_exposure_num")
seven_precise_exposure_num = layers.Input(shape=(1, ), name="seven_precise_exposure_num"), seven_precise_exposure_num = layers.Input(shape=(1, ), name="seven_precise_exposure_num")
fifteen_precise_exposure_num = layers.Input(shape=(1, ), name="fifteen_precise_exposure_num"), fifteen_precise_exposure_num = layers.Input(shape=(1, ), name="fifteen_precise_exposure_num")
thirty_precise_exposure_num = layers.Input(shape=(1, ), name="thirty_precise_exposure_num"), thirty_precise_exposure_num = layers.Input(shape=(1, ), name="thirty_precise_exposure_num")
sixty_precise_exposure_num = layers.Input(shape=(1, ), name="sixty_precise_exposure_num"), sixty_precise_exposure_num = layers.Input(shape=(1, ), name="sixty_precise_exposure_num")
ninety_precise_exposure_num = layers.Input(shape=(1, ), name="ninety_precise_exposure_num"), ninety_precise_exposure_num = layers.Input(shape=(1, ), name="ninety_precise_exposure_num")
history_precise_exposure_num = layers.Input(shape=(1, ), name="history_precise_exposure_num"), history_precise_exposure_num = layers.Input(shape=(1, ), name="history_precise_exposure_num")
one_vote_user_num = layers.Input(shape=(1, ), name="one_vote_user_num"), one_vote_user_num = layers.Input(shape=(1, ), name="one_vote_user_num")
three_vote_user_num = layers.Input(shape=(1, ), name="three_vote_user_num"), three_vote_user_num = layers.Input(shape=(1, ), name="three_vote_user_num")
seven_vote_user_num = layers.Input(shape=(1, ), name="seven_vote_user_num"), seven_vote_user_num = layers.Input(shape=(1, ), name="seven_vote_user_num")
fifteen_vote_user_num = layers.Input(shape=(1, ), name="fifteen_vote_user_num"), fifteen_vote_user_num = layers.Input(shape=(1, ), name="fifteen_vote_user_num")
thirty_vote_user_num = layers.Input(shape=(1, ), name="thirty_vote_user_num"), thirty_vote_user_num = layers.Input(shape=(1, ), name="thirty_vote_user_num")
sixty_vote_user_num = layers.Input(shape=(1, ), name="sixty_vote_user_num"), sixty_vote_user_num = layers.Input(shape=(1, ), name="sixty_vote_user_num")
ninety_vote_user_num = layers.Input(shape=(1, ), name="ninety_vote_user_num"), ninety_vote_user_num = layers.Input(shape=(1, ), name="ninety_vote_user_num")
history_vote_user_num = layers.Input(shape=(1, ), name="history_vote_user_num"), history_vote_user_num = layers.Input(shape=(1, ), name="history_vote_user_num")
one_reply_user_num = layers.Input(shape=(1, ), name="one_reply_user_num"), one_reply_user_num = layers.Input(shape=(1, ), name="one_reply_user_num")
three_reply_user_num = layers.Input(shape=(1, ), name="three_reply_user_num"), three_reply_user_num = layers.Input(shape=(1, ), name="three_reply_user_num")
seven_reply_user_num = layers.Input(shape=(1, ), name="seven_reply_user_num"), seven_reply_user_num = layers.Input(shape=(1, ), name="seven_reply_user_num")
fifteen_reply_user_num = layers.Input(shape=(1, ), name="fifteen_reply_user_num"), fifteen_reply_user_num = layers.Input(shape=(1, ), name="fifteen_reply_user_num")
thirty_reply_user_num = layers.Input(shape=(1, ), name="thirty_reply_user_num"), thirty_reply_user_num = layers.Input(shape=(1, ), name="thirty_reply_user_num")
sixty_reply_user_num = layers.Input(shape=(1, ), name="sixty_reply_user_num"), sixty_reply_user_num = layers.Input(shape=(1, ), name="sixty_reply_user_num")
ninety_reply_user_num = layers.Input(shape=(1, ), name="ninety_reply_user_num"), ninety_reply_user_num = layers.Input(shape=(1, ), name="ninety_reply_user_num")
history_reply_user_num = layers.Input(shape=(1, ), name="history_reply_user_num"), history_reply_user_num = layers.Input(shape=(1, ), name="history_reply_user_num")
one_browse_user_num = layers.Input(shape=(1, ), name="one_browse_user_num"), one_browse_user_num = layers.Input(shape=(1, ), name="one_browse_user_num")
three_browse_user_num = layers.Input(shape=(1, ), name="three_browse_user_num"), three_browse_user_num = layers.Input(shape=(1, ), name="three_browse_user_num")
seven_browse_user_num = layers.Input(shape=(1, ), name="seven_browse_user_num"), seven_browse_user_num = layers.Input(shape=(1, ), name="seven_browse_user_num")
fifteen_browse_user_num = layers.Input(shape=(1, ), name="fifteen_browse_user_num"), fifteen_browse_user_num = layers.Input(shape=(1, ), name="fifteen_browse_user_num")
thirty_browse_user_num = layers.Input(shape=(1, ), name="thirty_browse_user_num"), thirty_browse_user_num = layers.Input(shape=(1, ), name="thirty_browse_user_num")
sixty_browse_user_num = layers.Input(shape=(1, ), name="sixty_browse_user_num"), sixty_browse_user_num = layers.Input(shape=(1, ), name="sixty_browse_user_num")
ninety_browse_user_num = layers.Input(shape=(1, ), name="ninety_browse_user_num"), ninety_browse_user_num = layers.Input(shape=(1, ), name="ninety_browse_user_num")
history_browse_user_num = layers.Input(shape=(1, ), name="history_browse_user_num"), history_browse_user_num = layers.Input(shape=(1, ), name="history_browse_user_num")
one_reply_num = layers.Input(shape=(1, ), name="one_reply_num"), one_reply_num = layers.Input(shape=(1, ), name="one_reply_num")
three_reply_num = layers.Input(shape=(1, ), name="three_reply_num"), three_reply_num = layers.Input(shape=(1, ), name="three_reply_num")
seven_reply_num = layers.Input(shape=(1, ), name="seven_reply_num"), seven_reply_num = layers.Input(shape=(1, ), name="seven_reply_num")
fifteen_reply_num = layers.Input(shape=(1, ), name="fifteen_reply_num"), fifteen_reply_num = layers.Input(shape=(1, ), name="fifteen_reply_num")
thirty_reply_num = layers.Input(shape=(1, ), name="thirty_reply_num"), thirty_reply_num = layers.Input(shape=(1, ), name="thirty_reply_num")
sixty_reply_num = layers.Input(shape=(1, ), name="sixty_reply_num"), sixty_reply_num = layers.Input(shape=(1, ), name="sixty_reply_num")
ninety_reply_num = layers.Input(shape=(1, ), name="ninety_reply_num"), ninety_reply_num = layers.Input(shape=(1, ), name="ninety_reply_num")
history_reply_num = layers.Input(shape=(1, ), name="history_reply_num"), history_reply_num = layers.Input(shape=(1, ), name="history_reply_num")
one_ctr = layers.Input(shape=(1, ), name="one_ctr"), one_ctr = layers.Input(shape=(1, ), name="one_ctr")
three_ctr = layers.Input(shape=(1, ), name="three_ctr"), three_ctr = layers.Input(shape=(1, ), name="three_ctr")
seven_ctr = layers.Input(shape=(1, ), name="seven_ctr"), seven_ctr = layers.Input(shape=(1, ), name="seven_ctr")
fifteen_ctr = layers.Input(shape=(1, ), name="fifteen_ctr"), fifteen_ctr = layers.Input(shape=(1, ), name="fifteen_ctr")
thirty_ctr = layers.Input(shape=(1, ), name="thirty_ctr"), thirty_ctr = layers.Input(shape=(1, ), name="thirty_ctr")
sixty_ctr = layers.Input(shape=(1, ), name="sixty_ctr"), sixty_ctr = layers.Input(shape=(1, ), name="sixty_ctr")
ninety_ctr = layers.Input(shape=(1, ), name="ninety_ctr"), ninety_ctr = layers.Input(shape=(1, ), name="ninety_ctr")
history_ctr = layers.Input(shape=(1, ), name="history_ctr"), history_ctr = layers.Input(shape=(1, ), name="history_ctr")
one_vote_pure_rate = layers.Input(shape=(1, ), name="one_vote_pure_rate"), one_vote_pure_rate = layers.Input(shape=(1, ), name="one_vote_pure_rate")
three_vote_pure_rate = layers.Input(shape=(1, ), name="three_vote_pure_rate"), three_vote_pure_rate = layers.Input(shape=(1, ), name="three_vote_pure_rate")
seven_vote_pure_rate = layers.Input(shape=(1, ), name="seven_vote_pure_rate"), seven_vote_pure_rate = layers.Input(shape=(1, ), name="seven_vote_pure_rate")
fifteen_vote_pure_rate = layers.Input(shape=(1, ), name="fifteen_vote_pure_rate"), fifteen_vote_pure_rate = layers.Input(shape=(1, ), name="fifteen_vote_pure_rate")
thirty_vote_pure_rate = layers.Input(shape=(1, ), name="thirty_vote_pure_rate"), thirty_vote_pure_rate = layers.Input(shape=(1, ), name="thirty_vote_pure_rate")
sixty_vote_pure_rate = layers.Input(shape=(1, ), name="sixty_vote_pure_rate"), sixty_vote_pure_rate = layers.Input(shape=(1, ), name="sixty_vote_pure_rate")
ninety_vote_pure_rate = layers.Input(shape=(1, ), name="ninety_vote_pure_rate"), ninety_vote_pure_rate = layers.Input(shape=(1, ), name="ninety_vote_pure_rate")
history_vote_pure_rate = layers.Input(shape=(1, ), name="history_vote_pure_rate"), history_vote_pure_rate = layers.Input(shape=(1, ), name="history_vote_pure_rate")
one_reply_pure_rate = layers.Input(shape=(1, ), name="one_reply_pure_rate"), one_reply_pure_rate = layers.Input(shape=(1, ), name="one_reply_pure_rate")
three_reply_pure_rate = layers.Input(shape=(1, ), name="three_reply_pure_rate"), three_reply_pure_rate = layers.Input(shape=(1, ), name="three_reply_pure_rate")
seven_reply_pure_rate = layers.Input(shape=(1, ), name="seven_reply_pure_rate"), seven_reply_pure_rate = layers.Input(shape=(1, ), name="seven_reply_pure_rate")
fifteen_reply_pure_rate = layers.Input(shape=(1, ), name="fifteen_reply_pure_rate"), fifteen_reply_pure_rate = layers.Input(shape=(1, ), name="fifteen_reply_pure_rate")
thirty_reply_pure_rate = layers.Input(shape=(1, ), name="thirty_reply_pure_rate"), thirty_reply_pure_rate = layers.Input(shape=(1, ), name="thirty_reply_pure_rate")
sixty_reply_pure_rate = layers.Input(shape=(1, ), name="sixty_reply_pure_rate"), sixty_reply_pure_rate = layers.Input(shape=(1, ), name="sixty_reply_pure_rate")
ninety_reply_pure_rate = layers.Input(shape=(1, ), name="ninety_reply_pure_rate"), ninety_reply_pure_rate = layers.Input(shape=(1, ), name="ninety_reply_pure_rate")
history_reply_pure_rate = layers.Input(shape=(1, ), name="history_reply_pure_rate"), history_reply_pure_rate = layers.Input(shape=(1, ), name="history_reply_pure_rate")
card_fd = layers.Input(shape=(1, ), name="card_fd"), card_fd = layers.Input(shape=(1, ), name="card_fd")
card_sd = layers.Input(shape=(1, ), name="card_sd"), card_sd = layers.Input(shape=(1, ), name="card_sd")
card_fs = layers.Input(shape=(1, ), name="card_fs"), card_fs = layers.Input(shape=(1, ), name="card_fs")
card_ss = layers.Input(shape=(1, ), name="card_ss"), card_ss = layers.Input(shape=(1, ), name="card_ss")
card_fp = layers.Input(shape=(1, ), name="card_fp"), card_fp = layers.Input(shape=(1, ), name="card_fp")
card_sp = layers.Input(shape=(1, ), name="card_sp"), card_sp = layers.Input(shape=(1, ), name="card_sp")
card_p = layers.Input(shape=(1, ), name="card_p"), card_p = layers.Input(shape=(1, ), name="card_p")
card_fd2 = layers.Input(shape=(1, ), name="card_fd2"), card_fd2 = layers.Input(shape=(1, ), name="card_fd2")
card_sd2 = layers.Input(shape=(1, ), name="card_sd2"), card_sd2 = layers.Input(shape=(1, ), name="card_sd2")
card_fs2 = layers.Input(shape=(1, ), name="card_fs2"), card_fs2 = layers.Input(shape=(1, ), name="card_fs2")
card_ss2 = layers.Input(shape=(1, ), name="card_ss2"), card_ss2 = layers.Input(shape=(1, ), name="card_ss2")
card_fp2 = layers.Input(shape=(1, ), name="card_fp2"), card_fp2 = layers.Input(shape=(1, ), name="card_fp2")
card_sp2 = layers.Input(shape=(1, ), name="card_sp2"), card_sp2 = layers.Input(shape=(1, ), name="card_sp2")
card_p2 = layers.Input(shape=(1, ), name="card_p2"), card_p2 = layers.Input(shape=(1, ), name="card_p2")
# user tower # user tower
device_vector = layers.concatenate([ device_vector = layers.concatenate([
...@@ -397,7 +397,7 @@ def tractate_dssm_model(): ...@@ -397,7 +397,7 @@ def tractate_dssm_model():
layers.Embedding(DEVICE_DIM_DICT.get("device_ss2"), int(DEVICE_DIM_DICT.get("device_ss2") / 10))(device_ss2), layers.Embedding(DEVICE_DIM_DICT.get("device_ss2"), int(DEVICE_DIM_DICT.get("device_ss2") / 10))(device_ss2),
layers.Embedding(DEVICE_DIM_DICT.get("device_fp2"), int(DEVICE_DIM_DICT.get("device_fp2") / 10))(device_fp2), layers.Embedding(DEVICE_DIM_DICT.get("device_fp2"), int(DEVICE_DIM_DICT.get("device_fp2") / 10))(device_fp2),
layers.Embedding(DEVICE_DIM_DICT.get("device_sp"), int(DEVICE_DIM_DICT.get("device_sp2") / 10))(device_sp2), layers.Embedding(DEVICE_DIM_DICT.get("device_sp"), int(DEVICE_DIM_DICT.get("device_sp2") / 10))(device_sp2),
layers.Embedding(DEVICE_DIM_DICT.get("device_p2"), int(DEVICE_DIM_DICT.get("device_p2") / 10))(device_p2), layers.Embedding(DEVICE_DIM_DICT.get("device_p2"), int(DEVICE_DIM_DICT.get("device_p2") / 10))(device_p2)
]) ])
device_vector = layers.Dense(3000, activation=activations.relu)(device_vector) device_vector = layers.Dense(3000, activation=activations.relu)(device_vector)
device_vector = layers.Dense( device_vector = layers.Dense(
...@@ -410,7 +410,7 @@ def tractate_dssm_model(): ...@@ -410,7 +410,7 @@ def tractate_dssm_model():
# item tower # item tower
tractate_vector = layers.concatenate( tractate_vector = layers.concatenate(
[ [
layers.Embedding(TRACTATE_DIM_DICT.get("card_id"), 3000)(card_id), layers.Embedding(TRACTATE_DIM_DICT.get("card_id"), 300)(card_id),
layers.Embedding(TRACTATE_DIM_DICT.get("is_pure_author"), layers.Embedding(TRACTATE_DIM_DICT.get("is_pure_author"),
int(TRACTATE_DIM_DICT.get("is_pure_author") / 10))(is_pure_author), int(TRACTATE_DIM_DICT.get("is_pure_author") / 10))(is_pure_author),
layers.Embedding(TRACTATE_DIM_DICT.get("is_have_pure_reply"), layers.Embedding(TRACTATE_DIM_DICT.get("is_have_pure_reply"),
...@@ -586,7 +586,7 @@ def tractate_dssm_model(): ...@@ -586,7 +586,7 @@ def tractate_dssm_model():
layers.Embedding(TRACTATE_DIM_DICT.get("card_ss2"), int(TRACTATE_DIM_DICT.get("card_ss2") / 10))(card_ss2), layers.Embedding(TRACTATE_DIM_DICT.get("card_ss2"), int(TRACTATE_DIM_DICT.get("card_ss2") / 10))(card_ss2),
layers.Embedding(TRACTATE_DIM_DICT.get("card_fp2"), int(TRACTATE_DIM_DICT.get("card_fp2") / 10))(card_fp2), layers.Embedding(TRACTATE_DIM_DICT.get("card_fp2"), int(TRACTATE_DIM_DICT.get("card_fp2") / 10))(card_fp2),
layers.Embedding(TRACTATE_DIM_DICT.get("card_sp2"), int(TRACTATE_DIM_DICT.get("card_sp2") / 10))(card_sp2), layers.Embedding(TRACTATE_DIM_DICT.get("card_sp2"), int(TRACTATE_DIM_DICT.get("card_sp2") / 10))(card_sp2),
layers.Embedding(TRACTATE_DIM_DICT.get("card_p2"), int(TRACTATE_DIM_DICT.get("card_p2") / 10))(card_p2), layers.Embedding(TRACTATE_DIM_DICT.get("card_p2"), int(TRACTATE_DIM_DICT.get("card_p2") / 10))(card_p2)
]) ])
tractate_vector = layers.Dense(3000, activation=activations.relu)(tractate_vector) tractate_vector = layers.Dense(3000, activation=activations.relu)(tractate_vector)
......
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