{ "cells": [ { "cell_type": "markdown", "metadata": { "id": "kgMvP3SF-w_X" }, "source": [ "##### Copyright 2022 The TensorFlow Authors." ] }, { "cell_type": "code", "execution_count": 1, "metadata": { "cellView": "form", "execution": { "iopub.execute_input": "2024-08-24T11:22:50.377868Z", "iopub.status.busy": "2024-08-24T11:22:50.377364Z", "iopub.status.idle": "2024-08-24T11:22:50.381283Z", "shell.execute_reply": "2024-08-24T11:22:50.380753Z" }, "id": "yhrhZl5t-yUe" }, "outputs": [], "source": [ "#@title Licensed under the Apache License, Version 2.0 (the \"License\");\n", "# you may not use this file except in compliance with the License.\n", "# You may obtain a copy of the License at\n", "#\n", "# https://www.apache.org/licenses/LICENSE-2.0\n", "#\n", "# Unless required by applicable law or agreed to in writing, software\n", "# distributed under the License is distributed on an \"AS IS\" BASIS,\n", "# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n", "# See the License for the specific language governing permissions and\n", "# limitations under the License." ] }, { "cell_type": "markdown", "metadata": { "id": "ZuQzYr8B1J1K" }, "source": [ "# Visualizing TensorFlow Decision Forest Trees with dtreeviz\n", "\n", "
\n",
" ![]() | \n",
" \n",
" ![]() | \n",
" \n",
" ![]() | \n",
" \n",
" ![]() | \n",
"
\n", " | species | \n", "island | \n", "bill_length_mm | \n", "bill_depth_mm | \n", "flipper_length_mm | \n", "body_mass_g | \n", "sex | \n", "year | \n", "
---|---|---|---|---|---|---|---|---|
0 | \n", "Adelie | \n", "Torgersen | \n", "39.1 | \n", "18.7 | \n", "181.0 | \n", "3750.0 | \n", "male | \n", "2007 | \n", "
1 | \n", "Adelie | \n", "Torgersen | \n", "39.5 | \n", "17.4 | \n", "186.0 | \n", "3800.0 | \n", "female | \n", "2007 | \n", "
2 | \n", "Adelie | \n", "Torgersen | \n", "40.3 | \n", "18.0 | \n", "195.0 | \n", "3250.0 | \n", "female | \n", "2007 | \n", "
\n", " | species | \n", "island | \n", "bill_length_mm | \n", "bill_depth_mm | \n", "flipper_length_mm | \n", "body_mass_g | \n", "sex | \n", "year | \n", "
---|---|---|---|---|---|---|---|---|
0 | \n", "0 | \n", "Torgersen | \n", "39.1 | \n", "18.7 | \n", "181.0 | \n", "3750.0 | \n", "male | \n", "2007 | \n", "
1 | \n", "0 | \n", "Torgersen | \n", "39.5 | \n", "17.4 | \n", "186.0 | \n", "3800.0 | \n", "female | \n", "2007 | \n", "
2 | \n", "0 | \n", "Torgersen | \n", "40.3 | \n", "18.0 | \n", "195.0 | \n", "3250.0 | \n", "female | \n", "2007 | \n", "
\n", " | bill_depth_mm | \n", "bill_length_mm | \n", "body_mass_g | \n", "flipper_length_mm | \n", "island | \n", "sex | \n", "year | \n", "
---|---|---|---|---|---|---|---|
count | \n", "31.0 | \n", "31.0 | \n", "31.0 | \n", "31.0 | \n", "31 | \n", "31 | \n", "31.0 | \n", "
unique | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "1 | \n", "2 | \n", "<NA> | \n", "
top | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "Dream | \n", "female | \n", "<NA> | \n", "
freq | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "31 | \n", "19 | \n", "<NA> | \n", "
mean | \n", "18.090323 | \n", "37.26129 | \n", "3595.967742 | \n", "189.0 | \n", "NaN | \n", "NaN | \n", "2008.032258 | \n", "
std | \n", "1.216924 | \n", "2.002778 | \n", "471.173039 | \n", "6.21289 | \n", "NaN | \n", "NaN | \n", "0.836017 | \n", "
min | \n", "15.5 | \n", "32.1 | \n", "2900.0 | \n", "178.0 | \n", "NaN | \n", "NaN | \n", "2007.0 | \n", "
25% | \n", "17.2 | \n", "36.25 | \n", "3275.0 | \n", "185.0 | \n", "NaN | \n", "NaN | \n", "2007.0 | \n", "
50% | \n", "18.1 | \n", "37.0 | \n", "3500.0 | \n", "189.0 | \n", "NaN | \n", "NaN | \n", "2008.0 | \n", "
75% | \n", "18.65 | \n", "38.95 | \n", "3925.0 | \n", "193.0 | \n", "NaN | \n", "NaN | \n", "2009.0 | \n", "
max | \n", "21.1 | \n", "40.3 | \n", "4650.0 | \n", "202.0 | \n", "NaN | \n", "NaN | \n", "2009.0 | \n", "
\n", " | Type | \n", "LongestShell | \n", "Diameter | \n", "Height | \n", "WholeWeight | \n", "ShuckedWeight | \n", "VisceraWeight | \n", "ShellWeight | \n", "Rings | \n", "
---|---|---|---|---|---|---|---|---|---|
0 | \n", "M | \n", "0.455 | \n", "0.365 | \n", "0.095 | \n", "0.5140 | \n", "0.2245 | \n", "0.1010 | \n", "0.15 | \n", "15 | \n", "
1 | \n", "M | \n", "0.350 | \n", "0.265 | \n", "0.090 | \n", "0.2255 | \n", "0.0995 | \n", "0.0485 | \n", "0.07 | \n", "7 | \n", "
2 | \n", "F | \n", "0.530 | \n", "0.420 | \n", "0.135 | \n", "0.6770 | \n", "0.2565 | \n", "0.1415 | \n", "0.21 | \n", "9 | \n", "
\n", " | Diameter | \n", "Height | \n", "LongestShell | \n", "ShellWeight | \n", "ShuckedWeight | \n", "Type | \n", "VisceraWeight | \n", "WholeWeight | \n", "
---|---|---|---|---|---|---|---|---|
count | \n", "672.0 | \n", "672.0 | \n", "672.0 | \n", "672.0 | \n", "672.0 | \n", "672 | \n", "672.0 | \n", "672.0 | \n", "
unique | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "3 | \n", "<NA> | \n", "<NA> | \n", "
top | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "F | \n", "<NA> | \n", "<NA> | \n", "
freq | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "<NA> | \n", "328 | \n", "<NA> | \n", "<NA> | \n", "
mean | \n", "0.514509 | \n", "0.178088 | \n", "0.652247 | \n", "0.412751 | \n", "0.629519 | \n", "NaN | \n", "0.313077 | \n", "1.445691 | \n", "
std | \n", "0.033435 | \n", "0.016295 | \n", "0.040379 | \n", "0.08141 | \n", "0.150842 | \n", "NaN | \n", "0.070733 | \n", "0.280846 | \n", "
min | \n", "0.37 | \n", "0.125 | \n", "0.51 | \n", "0.3195 | \n", "0.4145 | \n", "NaN | \n", "0.1255 | \n", "0.9585 | \n", "
25% | \n", "0.495 | \n", "0.165 | \n", "0.625 | \n", "0.35075 | \n", "0.519 | \n", "NaN | \n", "0.264875 | \n", "1.246375 | \n", "
50% | \n", "0.51 | \n", "0.175 | \n", "0.65 | \n", "0.39375 | \n", "0.60175 | \n", "NaN | \n", "0.304 | \n", "1.37925 | \n", "
75% | \n", "0.535 | \n", "0.19 | \n", "0.675 | \n", "0.455 | \n", "0.702375 | \n", "NaN | \n", "0.354 | \n", "1.5985 | \n", "
max | \n", "0.63 | \n", "0.21 | \n", "0.8 | \n", "1.005 | \n", "1.2455 | \n", "NaN | \n", "0.59 | \n", "2.55 | \n", "