|View source on GitHub|
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
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.
class AdvancedArguments: Advanced control of the model that most users won't need to use.
class CartModel: Cart learning algorithm.
class DistributedGradientBoostedTreesModel: Distributed Gradient Boosted Trees learning algorithm.
class GradientBoostedTreesModel: Gradient Boosted Trees learning algorithm.
class RandomForestModel: Random Forest learning algorithm.