New! Use Simple ML for Sheets to apply machine learning to the data in your Google Sheets Read More

tfdf.tuner.Tuner

Stay organized with collections Save and categorize content based on your preferences.

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.