ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfdf.keras.wrappers

Wrapper around each learning algorithm.

This file is generated automatically by running the following commands: bazel build -c opt //third_party/tensorflow_decision_forests/keras:wrappers bazel-bin/third_party/tensorflow_decision_forests/keras/wrappers_wrapper_main
> third_party/tensorflow_decision_forests/keras/wrappers_pre_generated.py

Please don't change this file directly. Instead, changes the source. The documentation source is contained in the "GetGenericHyperParameterSpecification" method of each learner e.g. GetGenericHyperParameterSpecification in learner/gradient_boosted_trees/gradient_boosted_trees.cc contains the documentation (and meta-data) used to generate this file.

Classes

class AdvancedArguments: Advanced control of the model that most users won't need to use.

class CartModel: Cart learning algorithm.

class GradientBoostedTreesModel: Gradient Boosted Trees learning algorithm.

class RandomForestModel: Random Forest learning algorithm.

TaskType 'abstract_model_pb2.Task'