Skip to content

Commit dce42be

Browse files
author
shixiaowen03
committed
fix feature bias error
1 parent f0dd6e8 commit dce42be

File tree

2 files changed

+43
-46
lines changed

2 files changed

+43
-46
lines changed

.idea/workspace.xml

Lines changed: 42 additions & 45 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

recommendation/Basic-DeepFM-model/DeepFM-StepByStep.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -859,7 +859,7 @@
859859
"\n",
860860
"\n",
861861
"\"\"\"fm part\"\"\"\n",
862-
"fm_first_order = tf.nn.embedding_lookup(weights['feature_embeddings'],feat_index)\n",
862+
"fm_first_order = tf.nn.embedding_lookup(weights['feature_bias'],feat_index)\n",
863863
"fm_first_order = tf.reduce_sum(tf.multiply(fm_first_order,reshaped_feat_value),2)\n",
864864
"\n",
865865
"summed_features_emb = tf.reduce_sum(embeddings,1)\n",

0 commit comments

Comments
 (0)