Have a question? Connect with the community at the TensorFlow Forum Visit Forum

tfdf.builder.AbstractDecisionForestBuilder

Generic decision forest model builder.

Inherits From: AbstractBuilder

dataspec Dataspec, possibly partially constructed.

Can be called before close for advanced model edition.

objective Objective of the model.

Methods

add_tree

View source

Adds one tree to the model.

check_leaf

View source

Called on all the leaf nodes during the export.

check_non_leaf

View source

Called on all the non-leaf nodes during the export.

close

View source

Finalize the builder work.

This method should be called last.

model_type

View source

Unique key describing the type of the model.

observe_feature

View source

Register a feature and some of its possible value.

Generally, users don't need to call this function. An example of advanced exception is if a model does not refer to a specific possible categorical value, and if this value should be threader differently that the Out-of-vocabulary.

Should be called at least once on each of the model input features. If called multiple times with categorical_values the set of possible values will be the union of the categorical_valuess.

Args
feature Definition of the feature.
categorical_values Set of observed values. Only for categorical-like features.

specialized_header

View source

Gets the specialized header of the model.

specialized_header_filename

View source

Gets the filename of the specialized header.

yggdrasil_model_path

View source

Gets the path to the destination yggdrasil model.