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

tfdf.builder.AbstractBuilder

Generic model builder.

dataspec Dataspec, possibly partially constructed.

Can be called before close for advanced model edition.

objective Objective of the model.

Methods

close

View source

Finalize the builder work.

This method should be called last.

get_dictionary

View source

Gets the dictionary of a categorical(-set) string feature.

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 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 categorical_valuess.

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

set_dictionary

View source

Sets the dictionary of a categorical or categorical-set column.

yggdrasil_model_path

View source

Gets the path to the destination yggdrasil model.