Skip to content

Commit e2026ee

Browse files
committed
Update to Movielens
1 parent 5f583d6 commit e2026ee

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Experimental/Movielens Recommendation.ipynb

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -548,6 +548,8 @@
548548
" loss = tf.losses.mean_squared_error(labels, tf.squeeze(logits))\n",
549549
" return tf.estimator.EstimatorSpec(\n",
550550
" mode=mode,\n",
551+
" predictions=predictions,\n",
552+
" export_outputs=export_outputs,\n",
551553
" loss=loss,\n",
552554
" train_op=train_op\n",
553555
" )"

0 commit comments

Comments
 (0)