Skip to content

Base Class




The nodes in a network connecting the blocks.




The base class for different Block.

The Block can be connected together to build the search space for an AutoModel. Notably, many args in the init function are defaults to be a tunable variable when not specified by the user.


build, inputs=None)

Build the Block into a real Keras Model.

The subclasses should override this function and return the output node.


  • hp: HyperParameters. The hyperparameters for building the model.
  • inputs: A list of input node(s).


autokeras.Head(loss=None, metrics=None, **kwargs)

Base class for the heads, e.g. classification, regression.


  • loss Optional[Union[str, Callable, tensorflow.keras.losses.Loss]]: A Keras loss function. Defaults to None. If None, the loss will be inferred from the AutoModel.
  • metrics Optional[Union[List[Union[str, Callable, tensorflow.keras.metrics.Metric]], List[List[Union[str, Callable, tensorflow.keras.metrics.Metric]]], Dict[str, Union[str, Callable, tensorflow.keras.metrics.Metric]]]]: A list of Keras metrics. Defaults to None. If None, the metrics will be inferred from the AutoModel.