Skip to content

Releases: tensorflow/decision-forests

0.1.6

08 Jun 07:37
Compare
Choose a tag to compare

Features

  • Add hyper-parameter sorting_strategy to 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

26 May 14:48
Compare
Choose a tag to compare

Features

  • Raise an error of the number of classes is greater than 100 (can be disabled).
  • Raise an error if the model's task does not match the pd_dataframe_to_tf_dataset's task.

Bug fix

  • Fix failure when input feature contains commas.

0.1.4

21 May 18:52
Compare
Choose a tag to compare

Features

  • Stop the training when interrupting a colab cell / typing ctrl-c.
  • model.fit support training callbacks and a validation dataset.

Bug fix

  • Fix failure when there are not input features.

0.1.2

18 May 17:25
Compare
Choose a tag to compare

Features

  • Inference engines: QuickScorer Extended and Pred

0.1.0

17 May 12:43
Compare
Choose a tag to compare

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".