Google I / O là một kết quả hoàn hảo! Cập nhật các phiên TensorFlow Xem phiên

Lắp đặ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ụ thuộc bằng tay, bạn có thể sử dụng Docker TensorFlow xây dựng . Nếu bạn chọn 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: Quyết định TensorFlow rừng phụ thuộc vào rừng theo Quyết định Yggdrasil . Nếu bạn muốn chỉnh sửa mã Yggdrasil, bạn có thể sao chép các Yggdrasil github và thay đổi đường dẫn 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, chạy start_compile_docker.sh kịch bản và tiếp tục sang 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 cấu hình cho python3.8 và trình biên dịch mặc định trên máy tính 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 cần phải sử dụng cùng một phiên bản của Python như trong test_bazel.sh kịch bản.

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

Nếu cấu hình của bạn không tương thích với manylinux2010, một tổ chức phi manylinux2010 gói tương thích pip sẽ được sản xuất, và kiểm tra cuối cùng sẽ thất bại. 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ềulinux2010 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 các ví dụ trong examples/minimal.py . Các gói phần mềm Pip nằm ở dist/ thư mục.

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 ra các gói pip cho tất cả các phiên bản tương thích của python sử dụng pyenv bằng cách chạy lệnh sau đây. 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 (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ố