Skip to content

Commit c382638

Browse files
Neural-Link Teamtensorflow-copybara
authored andcommitted
Update pack_nbrs command to use max_nbrs = 3 for the IMDB-based graph-NSL tutorial.
PiperOrigin-RevId: 266561784
1 parent 94d2016 commit c382638

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

g3doc/tutorials/graph_keras_lstm_imdb.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -645,7 +645,7 @@
645645
"outputs": [],
646646
"source": [
647647
"!python -m neural_structured_learning.tools.pack_nbrs \\\n",
648-
"--max_nbrs=1 --add_undirected_edges=True \\\n",
648+
"--max_nbrs=3 --add_undirected_edges=True \\\n",
649649
"/tmp/imdb/train_data.tfr '' /tmp/imdb/graph_99.tsv \\\n",
650650
"/tmp/imdb/nsl_train_data.tfr"
651651
]

0 commit comments

Comments
 (0)