View source on GitHub
|
Wrapper around each learning algorithm.
This file is generated automatically by running the following commands:
bazel build -c opt //tensorflow_decision_forests/keras:wrappers
bazel-bin/tensorflow_decision_forests/keras/wrappers_wrapper_main
> 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 DistributedGradientBoostedTreesModel: Distributed Gradient Boosted Trees learning algorithm.
class GradientBoostedTreesModel: Gradient Boosted Trees learning algorithm.
class HyperparameterOptimizerModel: Hyperparameter Optimizer learning algorithm.
class MultiTaskItem: A single task in a multi-task configuration.
class MultitaskerModel: Multitasker learning algorithm.
class RandomForestModel: Random Forest learning algorithm.
Other Members | |
|---|---|
| TaskType |
'abstract_model_pb2.Task'
|