Update Recsys tutorial notebook (Temporal)

ylongqi · Oct 15, 2018 · 4371e1f · 4371e1f
1 parent 8e3d235
commit 4371e1f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tutorials/Temporal_aware_recommendation.ipynb b/tutorials/Temporal_aware_recommendation.ipynb
@@ -145,7 +145,9 @@
         "                            train=train, serve=serve)\n",
         "    \n",
         "    ## [TODO] Please define input ports (Hint: using \"ins\" parameter)\n",
-        "    @rec.traingraph.interactiongraph(ins=[?FILL IN?])\n",
+        "    ## @rec.traingraph.interactiongraph(ins=[?FILL IN?])\n",
+        "    ## Answer:\n",
+        "    @rec.traingraph.interactiongraph(ins=['seq_item_vec', 'seq_len', 'label'])\n",
         "    def f(subgraph):\n",
         "        RNNSoftmax(seq_item_vec=subgraph['seq_item_vec'], \n",
         "                   seq_len=subgraph['seq_len'], \n",
@@ -157,7 +159,9 @@
         "                   scope='RNNSoftmax')\n",
         "    \n",
         "    ## [TODO] Please define input ports (Hint: using \"ins\" parameter)\n",
-        "    @rec.servegraph.interactiongraph(ins=[?FILL IN?])\n",
+        "    ## @rec.servegraph.interactiongraph(ins=[?FILL IN?])\n",
+        "    ## Answer:\n",
+        "    @rec.servegraph.interactiongraph(ins=['seq_item_vec', 'seq_len'])\n",
         "    def f(subgraph):\n",
         "        RNNSoftmax(seq_item_vec=subgraph['seq_item_vec'], \n",
         "                   seq_len=subgraph['seq_len'],\n",
@@ -167,7 +171,6 @@
         "                   subgraph=subgraph, \n",
         "                   scope='RNNSoftmax')\n",
         "    \n",
-        "    \n",
         "    return rec"
       ],
       "execution_count": 0,