Update pack_nbrs command to use max_nbrs = 3 for the IMDB-based graph…

…-NSL tutorial. PiperOrigin-RevId: 266561784
tensorflow · Aug 31, 2019 · c382638 · c382638
1 parent 94d2016
commit c382638
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/g3doc/tutorials/graph_keras_lstm_imdb.ipynb b/g3doc/tutorials/graph_keras_lstm_imdb.ipynb
@@ -645,7 +645,7 @@
       "outputs": [],
       "source": [
         "!python -m neural_structured_learning.tools.pack_nbrs \\\n",
-        "--max_nbrs=1 --add_undirected_edges=True \\\n",
+        "--max_nbrs=3 --add_undirected_edges=True \\\n",
         "/tmp/imdb/train_data.tfr '' /tmp/imdb/graph_99.tsv \\\n",
         "/tmp/imdb/nsl_train_data.tfr"
       ]