Have a question? Connect with the community at the TensorFlow Forum Visit Forum

Installation

Installation with pip

To install TensorFlow Decision Forests, run:

pip3 install tensorflow_decision_forests --upgrade

Then, check the installation with:

python3 -c "import tensorflow_decision_forests as tfdf; print('Found TF-DF v' + tfdf.__version__)"

Build from source

Requirements

  • Microsoft Visual Studio >= 2019 (Windows) or GCC>=7.3 (Linux).
  • Bazel >= 3.7.2
  • Python >= 3
  • Git
  • Python's numpy
  • MSYS2 (Windows only)

Download and compile TensorFlow Decision Forests as follow:

# Download the source code of TF-DF.
git clone https://github.com/tensorflow/decision-forests.git
cd decision-forests

# Download TensorFlow's BazelRC.
# See "tensorflow_bazelrc" for the compilation options.
# Note: If you compile with an older version of TensorFlow, download the
# corresponding bazelrc.
TENSORFLOW_BAZELRC="tensorflow_bazelrc"
wget https://raw.githubusercontent.com/tensorflow/tensorflow/master/.bazelrc -O ${TENSORFLOW_BAZELRC}

# Compile TensorFlow Decision Forests.
bazel-3.7.2 --bazelrc=${TENSORFLOW_BAZELRC} build //tensorflow_decision_forests/...:all --config=linux

# Create a pip package.
./tools/build_pip_package.sh simple

# Install the pip package
pip3 install dist/*.whl

Troubleshooting