Updates NSL tutorials to ensure they work in the open source environment.

PiperOrigin-RevId: 266400651

Author: Neural-Link Team

g3doc/tutorials/graph_keras_lstm_imdb.ipynb

@@ -45,10 +45,10 @@
         "\n",
         "\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_lstm_imdb.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_lstm_imdb\"\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_lstm_imdb.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/colab_logo_32px.png\" /\u003eRun in Google Colab\u003c/a\u003e\n",
+        "    \u003ca target=\"_blank\" href=\"https://colab.research.google.com/github/tensorflow/neural-structured-learning/blob/master/g3doc/tutorials/graph_keras_lstm_imdb.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/colab_logo_32px.png\" /\u003eRun in Google Colab\u003c/a\u003e\n",
         "  \u003c/td\u003e\n",
         "  \u003ctd\u003e\n",
         "    \u003ca target=\"_blank\" href=\"https://github.com/tensorflow/neural-structured-learning/blob/master/g3doc/tutorials/graph_keras_lstm_imdb.ipynb\"\u003e\u003cimg src=\"https://www.tensorflow.org/images/GitHub-Mark-32px.png\" /\u003eView source on GitHub\u003c/a\u003e\n",
@@ -114,8 +114,9 @@
       "source": [
         "## Requirements\n",
         "\n",
-        "1.  Install the Neural Structured Learning package.\n",
-        "2.  Install tensorflow-hub"
+        "1.  Install TensorFlow 2.x to create an interactive developing environment with eager execution.\n",
+        "2. Install the Neural Structured Learning package.\n",
+        "3. Install tensorflow-hub."
       ]
     },
     {
@@ -128,8 +129,9 @@
       },
       "outputs": [],
       "source": [
+        "!pip install --quiet tensorflow==2.0.0-rc0\n",
         "!pip install --quiet neural-structured-learning\n",
-        "!pip install --quit tensorflow-hub"
+        "!pip install --quiet tensorflow-hub"
       ]
     },
     {
@@ -162,14 +164,18 @@
         "\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",
-        "tf.keras.backend.clear_session()  # Resets notebook state\n",
         "\n",
         "import tensorflow_hub as hub\n",
         "\n",
-        "print(tf.__version__)"
+        "# Resets notebook state\n",
+        "tf.keras.backend.clear_session()\n",
+        "\n",
+        "print(\"Version: \", tf.__version__)\n",
+        "print(\"Eager mode: \", tf.executing_eagerly())\n",
+        "print(\"Hub version: \", hub.__version__)\n",
+        "print(\"GPU is\", \"available\" if tf.test.is_gpu_available() else \"NOT AVAILABLE\")"
       ]
     },
     {
@@ -436,6 +442,10 @@
       },
       "outputs": [],
       "source": [
+        "# This is necessary because hub.KerasLayer assumes tensor hashability, which\n",
+        "# is not supported in eager mode.\n",
+        "tf.compat.v1.disable_tensor_equality()\n",
+        "\n",
         "pretrained_embedding = 'https://tfhub.dev/google/tf2-preview/gnews-swivel-20dim/1'\n",
         "\n",
         "hub_layer = hub.KerasLayer(\n",
@@ -730,7 +740,7 @@
         "    self.num_classes = 2\n",
         "    self.max_seq_length = 256\n",
         "    self.vocab_size = 10000\n",
-        "    ### Neural Graph Learning parameters\n",
+        "    ### neural graph learning parameters\n",
         "    self.distance_type = nsl.configs.DistanceType.L2\n",
         "    self.graph_regularization_multiplier = 0.1\n",
         "    self.num_neighbors = 1\n",
@@ -1030,6 +1040,7 @@
         "validation_fraction = 0.9\n",
         "validation_size = int(validation_fraction *\n",
         "                      int(training_samples_count / HPARAMS.batch_size))\n",
+        "print(validation_size)\n",
         "validation_dataset = train_dataset.take(validation_size)\n",
         "train_dataset = train_dataset.skip(validation_size)"
       ]

g3doc/tutorials/graph_keras_mlp_cora.ipynb

@@ -109,38 +109,15 @@
         "\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."
+        "1.  Install TensorFlow 2.x to create an interactive developing environment with eager execution.\n",
+        "2.  Install the Neural Structured Learning package."
       ]
     },
     {
@@ -153,6 +130,7 @@
       },
       "outputs": [],
       "source": [
+        "!pip install --quiet tensorflow==2.0.0-rc0\n",
         "!pip install --quiet neural-structured-learning"
       ]
     },
@@ -183,7 +161,11 @@
         "import tensorflow as tf\n",
         "\n",
         "# Resets notebook state\n",
-        "tf.keras.backend.clear_session()"
+        "tf.keras.backend.clear_session()\n",
+        "\n",
+        "print(\"Version: \", tf.__version__)\n",
+        "print(\"Eager mode: \", tf.executing_eagerly())\n",
+        "print(\"GPU is\", \"available\" if tf.test.is_gpu_available() else \"NOT AVAILABLE\")"
       ]
     },
     {
@@ -324,7 +306,7 @@
         "id": "2gYWAqJqZ76I"
       },
       "source": [
-        "## Hyperparameters and constants\n",
+        "## Hyperparameters\n",
         "\n",
         "We will use an instance of `HParams` to inclue various hyperparameters and\n",
         "constants used for training and evaluation. We briefly describe each of them\n",