Skip to content

Commit f0dd6e8

Browse files
author
shixiaowen03
committed
transformer & deepfm step by step
1 parent ae33497 commit f0dd6e8

File tree

2 files changed

+26
-69
lines changed

2 files changed

+26
-69
lines changed

.idea/workspace.xml

Lines changed: 25 additions & 68 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
@@ -802,7 +802,7 @@
802802
"feat_index = tf.placeholder(tf.int32,shape=[None,None],name='feat_index')\n",
803803
"feat_value = tf.placeholder(tf.float32,shape=[None,None],name='feat_value')\n",
804804
"\n",
805-
"label = tf.placeholder(tf.float32,shape=[None],name='label')\n",
805+
"label = tf.placeholder(tf.float32,shape=[None,1],name='label')\n",
806806
"\n",
807807
"\"\"\"建立weights\"\"\"\n",
808808
"weights = dict()\n",

0 commit comments

Comments
 (0)