Cài đặt

Cài đặt với Pip

Cài đặt Rừng quyết định TensorFlow bằng cách chạy:

# Install TensorFlow Decision Forests.
pip3 install tensorflow_decision_forests --upgrade

Sau đó, kiểm tra cài đặt bằng cách chạy:

# Check the version of TensorFlow Decision Forests.
python3 -c "import tensorflow_decision_forests as tfdf; print('Found TF-DF v' + tfdf.__version__)"

Xây dựng từ nguồn

Linux

Thành lập

Yêu cầu

  • Bazel> = 3.7.2
  • Python> = 3
  • Git
  • Các gói Python: gấu trúc numpy tensorflow

Thay vì cài đặt các phần phụ thuộc bằng tay, bạn có thể sử dụng trình kết nối TensorFlow Build . Nếu bạn chọn các tùy chọn này, hãy cài đặt Docker:

Tổng hợp

Tải xuống TensorFlow Quyết định Rừng như sau:

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

Tùy chọn: Rừng Quyết định TensorFlow phụ thuộc vào Rừng Quyết định Yggdrasil . Nếu bạn muốn chỉnh sửa mã Yggdrasil, bạn có thể sao chép github Yggdrasil và thay đổi đường dẫn cho phù hợp trong third_party/yggdrasil_decision_forests/workspace.bzl .

Tùy chọn: Nếu bạn muốn sử dụng tùy chọn docker, hãy chạy tập lệnh start_compile_docker.sh và tiếp tục bước tiếp theo. Nếu bạn không muốn sử dụng tùy chọn docker, hãy tiếp tục trực tiếp bước tiếp theo.

# Optional: Install and start the build docker.
./tools/start_compile_docker.sh

Biên dịch và chạy các bài kiểm tra đơn vị của TF-DF bằng lệnh sau. Lưu ý rằng test_bazel.sh được định cấu hình cho python3.8 và trình biên dịch mặc định trên máy của bạn. Chỉnh sửa tệp trực tiếp để thay đổi cấu hình này.

# Build and test TF-DF.
./tools/test_bazel.sh

Tạo và kiểm tra một gói pip bằng lệnh sau. Thay thế python3.8 bằng phiên bản python bạn muốn sử dụng. Lưu ý rằng bạn không phải sử dụng cùng một phiên bản Python như trong tập lệnh test_bazel.sh .

Nếu cấu hình của bạn tương thích với manylinux2014 , một gói pip tương thích manylinux2014 sẽ được tạo.

Nếu cấu hình của bạn không tương thích với manylinux2014, một gói pip không tương thích với manylinux2014 sẽ được tạo và kiểm tra cuối cùng sẽ không thành công. Không thành vấn đề nếu bạn muốn sử dụng TF-DF trên máy của riêng mình. Một cách dễ dàng để làm cho bản dựng nhiềulinux2014 tương thích là sử dụng bộ công cụ được đề cập ở trên.

# Build and test a Pip package.
./tools/build_pip_package.sh python3.8

Lệnh này sẽ cài đặt gói pip TF-DF và chạy ví dụ trong example examples/minimal.py . Gói Pip nằm trong thư mục dist/ .

Nếu bạn muốn tạo gói Pip cho phiên bản Python tương thích khác, hãy chạy:

# Install the other versions of python (assume only python3.8 is installed; this is the case in the build docker).
sudo apt-get update && sudo apt-get install python3.7 python3.9 python3-pip

# Create the Pip package for the other version of python
./tools/build_pip_package.sh python3.7
./tools/build_pip_package.sh python3.9

Ngoài ra, bạn có thể tạo gói pip cho tất cả phiên bản tương thích của python bằng pyenv bằng cách chạy lệnh sau. Xem tiêu đề của tools/build_pip_package.sh để biết thêm chi tiết.

# Build and test all the Pip package using Pyenv.
./tools/build_pip_package.sh ALL_VERSIONS

Hệ điều hành Mac

Thành lập

Yêu cầu

  • Coreutils (được thử nghiệm với brew install coreutils )
  • Bazel> = 3.7.2
  • Python> = 3 (đã thử nghiệm với brew install python )
  • Git
  • JDK 11
  • Các gói Python: gấu trúc numpy tensorflow

Tổng hợp

Làm theo các bước tương tự như đối với biên dịch linux mà không có Docker.

Lưu ý cuối cùng

Việc biên dịch TF-DF dựa vào (kể từ ngày 17 tháng 12 năm 2021) vào gói TensorFlow Pip và sự phụ thuộc của TensorFlow Bazel. Một phần nhỏ của TensorFlow sẽ được biên dịch. Biên dịch TF-DF trên một máy trạm mạnh mẽ mất ~ 10 phút.

Xử lý sự cố