Hyper U-Net¶
-
class
src.pipelines.tensorflow_v2.models.hyper_unet.HyperUnet(input_shape, output_channels)[source]¶ A Hyper U-Net model which can be tuned by kerastuner
-
build(hp)[source]¶ A function which creates a kerastuner hyper model with a defined search space. This function signature matches the requirements of the hyperparameter tuning algorithms in kerastuner.
- Parameters
hp (
HyperParameters) – a HyperParameters instance- Return type
HyperModel- Returns
a HyperUnet instance
-