tfdf.tuner.Tuner

Abstract tuner class.

The user is expected to use one of its instances e.g. RandomSearch.

Methods

choice

View source

Adds a hyperparameter with a list of possible values.

Args
key Name of the hyper-parameter.
values List of possible value for the hyperparameter.
merge If false (default), raises an error if the hyper-parameter already exist. If true, adds values to the parameter if it already exist.

Returns
The conditional SearchSpace corresponding to the values in "values".

set_base_learner

View source

Sets the base learner key.

train_config

View source

YDF training configuration for the Hyperparameter optimizer.