TensorFlow 결정 포레스트

import tensorflow_decision_forests as tfdf
import pandas as pd

# Load a dataset in a Pandas dataframe.
train_df = pd.read_csv("project/train.csv")
test_df = pd.read_csv("project/test.csv")

# Convert the dataset into a TensorFlow dataset.
train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(train_df, label="my_label")
test_ds = tfdf.keras.pd_dataframe_to_tf_dataset(test_df, label="my_label")

# Train a Random Forest model.
model = tfdf.keras.RandomForestModel()
model.fit(train_ds)

# Summary of the model structure.
model.summary()

# Evaluate the model.
model.evaluate(test_ds)

# Export the model to a SavedModel.
model.save("project/model")

TensorFlow 결정 포레스트(TF-DF)는 결정 포레스트 모델의 학습, 제공, 해석을 위한 최첨단 알고리즘 컬렉션입니다. 라이브러리는 Keras 모델의 컬렉션이며 분류, 회귀, 순위 지정을 지원합니다.

TF-DF는 Yggdrasil 결정 포레스트 C++ 라이브러리의 래퍼입니다. TF-DF로 학습된 모델은 Yggdrasil 결정 포레스트 모델과 호환되며 반대의 경우도 마찬가지입니다.

하지만 TF-DF는 아직 Mac(#16) 또는 Windows(#3)에서 사용할 수 없으며, 이를 해결하기 위해 노력 중입니다.

키워드: 결정 포레스트, TensorFlow, 랜덤 포레스트, 경사 부스트된 트리, CART, 모델 해석

커뮤니티

사용할 수 있는 리소스는 다음과 같습니다.

참여

TensorFlow 결정 포레스트 및 Yggdrasil 결정 포레스트에 관한 참여를 환영합니다. 참여하려면 개발자 매뉴얼을 검토하세요.