Install tf_keras in notebooks

PiperOrigin-RevId: 602711911
tensorflow · Jan 30, 2024 · 14b5832 · 14b5832
1 parent d1d8d08
commit 14b5832
Show file tree

Hide file tree

Showing 6 changed files with 41 additions and 10 deletions.
diff --git a/documentation/tf_df_in_tf_js.md b/documentation/tf_df_in_tf_js.md
@@ -39,6 +39,7 @@ to TensorFlow.js.
 
 ```python
 !pip install tensorflow tensorflow_decision_forests 'tensorflowjs>=4.4.0'
+!pip install tf_keras
 
 # Prepare and load the model with TensorFlow
 import tensorflow as tf

diff --git a/documentation/tutorials/advanced_colab.ipynb b/documentation/tutorials/advanced_colab.ipynb
@@ -89,8 +89,10 @@
       },
       "outputs": [],
       "source": [
-        "# Install TensorFlow Dececision Forests.\n",
+        "# Install TensorFlow Decision Forests.\n",
         "!pip install tensorflow_decision_forests\n",
+        "# TF-DF requires Tensorflow < 2.15 or tf_keras\n",
+        "!pip install tf_keras\n",
         "\n",
         "# Use wurlitzer to show the training logs.\n",
         "!pip install wurlitzer"
@@ -104,12 +106,16 @@
       },
       "outputs": [],
       "source": [
+        "import os\n",
+        "# Keep using Keras 2\n",
+        "os.environ['TF_USE_LEGACY_KERAS'] = '1'\n",
+        "\n",
         "import tensorflow_decision_forests as tfdf\n",
         "\n",
-        "import os\n",
         "import numpy as np\n",
         "import pandas as pd\n",
         "import tensorflow as tf\n",
+        "import tf_keras\n",
         "import matplotlib.pyplot as plt\n",
         "import math\n",
         "import collections"

diff --git a/documentation/tutorials/beginner_colab.ipynb b/documentation/tutorials/beginner_colab.ipynb
@@ -108,7 +108,9 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tensorflow_decision_forests"
+        "!pip install tensorflow_decision_forests\n",
+        "# TF-DF requires Tensorflow < 2.15 or tf_keras\n",
+        "!pip install tf_keras\n"
       ]
     },
     {
@@ -148,12 +150,16 @@
       },
       "outputs": [],
       "source": [
+        "import os\n",
+        "# Keep using Keras 2\n",
+        "os.environ['TF_USE_LEGACY_KERAS'] = '1'\n",
+        "\n",
         "import tensorflow_decision_forests as tfdf\n",
         "\n",
-        "import os\n",
         "import numpy as np\n",
         "import pandas as pd\n",
         "import tensorflow as tf\n",
+        "import tf_keras\n",
         "import math"
       ]
     },

diff --git a/documentation/tutorials/intermediate_colab.ipynb b/documentation/tutorials/intermediate_colab.ipynb
@@ -96,7 +96,9 @@
       "outputs": [],
       "source": [
         "# Install TensorFlow Dececision Forests\n",
-        "!pip install tensorflow_decision_forests"
+        "!pip install tensorflow_decision_forests\n",
+        "# TF-DF requires Tensorflow < 2.15 or tf_keras\n",
+        "!pip install tf_keras\n"
       ]
     },
     {
@@ -136,12 +138,16 @@
       },
       "outputs": [],
       "source": [
+        "import os\n",
+        "# Keep using Keras 2\n",
+        "os.environ['TF_USE_LEGACY_KERAS'] = '1'\n",
+        "\n",
         "import tensorflow_decision_forests as tfdf\n",
         "\n",
-        "import os\n",
         "import numpy as np\n",
         "import pandas as pd\n",
         "import tensorflow as tf\n",
+        "import tf_keras\n",
         "import math"
       ]
     },

diff --git a/documentation/tutorials/model_composition_colab.ipynb b/documentation/tutorials/model_composition_colab.ipynb
@@ -217,7 +217,9 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tensorflow_decision_forests -U --quiet"
+        "!pip install tensorflow_decision_forests -U --quiet\n",
+        "# TF-DF requires Tensorflow < 2.15 or tf_keras\n",
+        "!pip install tf_keras -U --quiet\n"
       ]
     },
     {
@@ -258,12 +260,16 @@
       },
       "outputs": [],
       "source": [
+        "import os\n",
+        "# Keep using Keras 2\n",
+        "os.environ['TF_USE_LEGACY_KERAS'] = '1'\n",
+        "\n",
         "import tensorflow_decision_forests as tfdf\n",
         "\n",
-        "import os\n",
         "import numpy as np\n",
         "import pandas as pd\n",
         "import tensorflow as tf\n",
+        "import tf_keras\n",
         "import math\n",
         "import matplotlib.pyplot as plt"
       ]

diff --git a/documentation/tutorials/ranking_colab.ipynb b/documentation/tutorials/ranking_colab.ipynb
@@ -92,7 +92,9 @@
       },
       "outputs": [],
       "source": [
-        "!pip install tensorflow_decision_forests"
+        "!pip install tensorflow_decision_forests\n",
+        "# TF-DF requires Tensorflow < 2.15 or tf_keras\n",
+        "!pip install tf_keras\n"
       ]
     },
     {
@@ -132,12 +134,16 @@
       },
       "outputs": [],
       "source": [
+        "import os\n",
+        "# Keep using Keras 2\n",
+        "os.environ['TF_USE_LEGACY_KERAS'] = '1'\n",
+        "\n",
         "import tensorflow_decision_forests as tfdf\n",
         "\n",
-        "import os\n",
         "import numpy as np\n",
         "import pandas as pd\n",
         "import tensorflow as tf\n",
+        "import tf_keras\n",
         "import math"
       ]
     },