|View source on GitHub|
CART model builder.
tfdf.builder.CARTBuilder( path: str, objective:
tfdf.py_tree.objective.AbstractObjective, model_format: Optional[
tfdf.builder.ModelFormat] = tfdf.builder.ModelFormat.TENSORFLOW_SAVED_MODEL, winner_take_all: Optional[bool] = False, import_dataspec: Optional[data_spec_pb2.DataSpecification] = None )
A CART is represented as a Random Forest with one tree.
Dataspec, possibly partially constructed.
Can be called before
||Objective of the model.|
Adds one tree to the model.
Called on all the leaf nodes during the export.
Called on all the non-leaf nodes during the export.
Finalize the builder work.
This method should be called last.
get_dictionary( col_name: str ) -> List[str]
Gets the dictionary of a categorical(-set) string feature.
model_type() -> str
Unique key describing the type of the model.
tfdf.inspector.SimpleColumnSpec, categorical_values: Optional[Union[List[str], List[int]]] = None )
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 treated differently than out-of-vocabulary values.
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
||Definition of the feature.|
||Set of observed values. Only for categorical-like features.|
set_dictionary( col_name: str, dictionary: List[str] ) -> None
Sets the dictionary of a categorical or categorical-set column.
specialized_header() -> Any
Gets the specialized header of the model.
specialized_header_filename() -> str
Gets the filename of the specialized header.
Gets the path to the destination yggdrasil model.