install dtreeviz for colab

parrt · Oct 22, 2022 · ca2a2af · ca2a2af
1 parent d44612b
commit ca2a2af
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/notebooks/dtreeviz_tensorflow_visualisations.ipynb b/notebooks/dtreeviz_tensorflow_visualisations.ipynb
@@ -44,6 +44,9 @@
    "source": [
     "import sys\n",
     "import os\n",
+    "if 'google.colab' in sys.modules:\n",
+    "  !pip install -q dtreeviz\n",
+    "\n",
     "# add library module to PYTHONPATH\n",
     "sys.path.append(f\"{os.getcwd()}/../\")\n",
     "\n",