tfdf.tuner.SearchSpace

Set of hyperparameter and their respective possible values.

tfdf.tuner.SearchSpace(
    fields: Fields,
    parent_values: Optional[hyperparameter_pb2.HyperParameterSpace.DiscreteCandidates] = None
)

The user is not expected to create a "SearchSpace" object directly. Instead, SearchSpace object are instantiated by tuners.

Methods

choice(
    key: str,
    values: Union[List[int], List[float], List[str], List[bool]],
    merge: bool = False
) -> 'SearchSpace'

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