Updates the links and TF version in graph regularization colab.

PiperOrigin-RevId: 266203589

Author: Neural-Link Team

g3doc/tutorials/graph_keras_mlp_cora.ipynb

@@ -53,7 +53,7 @@
       "source": [
         "\u003ctable class=\"tfo-notebook-buttons\" align=\"left\"\u003e\n",
         "  \u003ctd\u003e\n",
-        "    \u003ca target=\"_blank\" href=\"https://www.tensorflow.org/neural_structured_learning/tutorials/graph_keras_mlp_cora.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/tf_logo_32px.png\" /\u003eView on TensorFlow.org\u003c/a\u003e\n",
+        "    \u003ca target=\"_blank\" href=\"https://www.tensorflow.org/neural_structured_learning/tutorials/graph_keras_mlp_cora\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/tf_logo_32px.png\" /\u003eView on TensorFlow.org\u003c/a\u003e\n",
         "  \u003c/td\u003e\n",
         "  \u003ctd\u003e\n",
         "    \u003ca target=\"_blank\" href=\"https://colab.research.google.com/github/tensorflow/neural-structured-learning/blob/master/g3doc/tutorials/graph_keras_mlp_cora.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/colab_logo_32px.png\" /\u003eRun in Google Colab\u003c/a\u003e\n",
@@ -106,7 +106,40 @@
       },
       "source": [
         "## Setup\n",
-        "\n",
+        "\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "colab_type": "text",
+        "id": "v_39VWm4yIgt"
+      },
+      "source": [
+        "Install Tensorflow 2.0 to create an interactive developing environment with\n",
+        "eager execution."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 0,
+      "metadata": {
+        "colab": {},
+        "colab_type": "code",
+        "id": "2m7Sp74HyFVS"
+      },
+      "outputs": [],
+      "source": [
+        "!pip install --quiet tensorflow==2.0.0-rc0"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "colab_type": "text",
+        "id": "hgSLDi0SyBuO"
+      },
+      "source": [
         "Install the Neural Structured Learning package."
       ]
     },
@@ -147,9 +180,7 @@
         "\n",
         "import neural_structured_learning as nsl\n",
         "\n",
-        "# Enable TensorFlow v2 behavior\n",
         "import tensorflow as tf\n",
-        "tf.compat.v1.enable_v2_behavior()\n",
         "\n",
         "# Resets notebook state\n",
         "tf.keras.backend.clear_session()"
@@ -244,7 +275,7 @@
       },
       "outputs": [],
       "source": [
-        "!wget https://github.com/tensorflow/neural-structured-learning/blob/master/examples/preprocess/cora/preprocess_cora_dataset.py\n",
+        "!wget https://raw.githubusercontent.com/tensorflow/neural-structured-learning/master/neural_structured_learning/examples/preprocess/cora/preprocess_cora_dataset.py\n",
         "\n",
         "!python preprocess_cora_dataset.py \\\n",
         "--input_cora_content=/tmp/cora/cora.content \\\n",

neural_structured_learning/examples/preprocess/cora/preprocess_cora_dataset.py

@@ -76,6 +76,9 @@
     """Input file for Cora content that contains ID, words and labels.""")
 flags.DEFINE_string('input_cora_graph', '/tmp/cora/cora.cites',
                     """Input file for Cora citation graph in TSV format.""")
+flags.DEFINE_integer(
+    'max_nbrs', None,
+    'The maximum number of neighbors to merge into each labeled Example.')
 flags.DEFINE_float(
     'train_percentage', 0.8,
     """The percentage of examples to be created as training data. The rest
@@ -158,8 +161,11 @@ def main(unused_argv):
   # neighbors for transductive learning purpose. In other words, the labels of
   # test_examples are not used.
   with tf.io.TFRecordWriter(FLAGS.output_train_data) as writer:
-    for merged_example in pack_nbrs.join_examples(train_examples, test_examples,
-                                                  graph, FLAGS.max_nbrs):
+    # Here we call a private function in pack_nbrs to join the examples. This is
+    # one-off for demonstration purpose only. Later on we will refactor that
+    # function to a public API.
+    for merged_example in pack_nbrs._join_examples(  # pylint: disable=protected-access
+        train_examples, test_examples, graph, FLAGS.max_nbrs):
       writer.write(merged_example.SerializeToString())
 
   logging.info('Output training data written to TFRecord file: %s.',