Unet#

class tfts.models.unet.Unet(predict_sequence_length: int = 1, config: UnetConfig | None = None)[source]#

Unet model for sequence-to-sequence prediction tasks.

Methods

`build_model`(inputs)
`compute_output_shape`(input_shape)
`get_config`()
`load_pretrained_weights`(weights_dir)
`predict`(x, **kwargs)
`save_model`(weights_dir)
`save_pretrained`(save_directory[, ...])
`save_weights`(weights_path)
`summary`()
`to_model`()