baby.training.hypermodels.UNet

class baby.training.hypermodels.UNet(input_shape, outputs, weights=None)

Bases: HyperModel

Attributes
defaults

Methods

build(hp)

Builds a model.

__init__(input_shape, outputs, weights=None)

Methods

__init__(input_shape, outputs[, weights])

build(hp)

Builds a model.

Attributes

defaults

build(hp)

Builds a model.

# Arguments:

hp: A HyperParameters instance.

# Returns:

A model instance.