Releases: tensorflow/decision-forests
Releases · tensorflow/decision-forests
0.1.8
Features
- Model can be composed with the functional Keras API before being trained.
- Makes all the Yggdrasil structural variable importances available.
- Makes getting the variable importance instantaneous.
- Surface the
nameargument in the model classes constructors. - Add a
postprocessingmodel constructor argument to easy apply
post-processing on the model predictions without relying on the Keras
Functional API. - Add
extract_all_treesmethod in the model inspector to efficiently exact
all the trees. - Add
num_threadsconstructor argument to control the number of training
threads without using the advanced configuration. - By default, remove the temporary directory used to train the model when the
model python object is garbage collected. - Add the
import_dataspecconstructor argument to the model builder to
import the feature definition and dictionaries (instead of relying on
automatic discovery).
Changes
- When saving a model in a directory already containing a model, only the
assetsdirectory is entirely removed before the export (instead of the
entire model directory).
Fixes
- Wrong label shape in the model inspector's objective field for
pre-integerized labels.
0.1.7
Features
- Add more of characters to the non-recommended list of feature name
characters. - Make the inference op multi-thread compatible.
- Print an explicit error and some instructions when training a model with a
Pandas dataframe. pd_dataframe_to_tf_datasetcan automatically rename feature to make them
compatible with SavedModel export signatures.model.save(...)can override an existing model.- The link function of GBT model can be removed. For example, a binary
classification GBT model trained with apply_link_function=False will output
logits.
0.1.6
Features
- Add hyper-parameter
sorting_strategyto disable the computation of the
pre-sorted index (slower to train, but consumes less memory). - Format wrapper code for colab help display.
- Raises an error when a feature name is not compatible (e.g. contains a space).
0.1.5
0.1.4
0.1.2
0.1.0
Release 0.1.0 (2021-05-11)
Initial Release of TensorFlow Decision Forests.
Features
- Random Forest learner.
- Gradient Boosted Tree learner.
- CART learner.
- Model inspector: Inspect the internal model structure.
- Model plotter: Plot decision trees.
- Model builder: Create model "by hand".