buddi.models.components.wrapped_branches#

Functions

Input([shape, batch_size, dtype, sparse, ...])

Used to instantiate a Keras tensor.

build_decoder_branch(sup_inputs, ...[, ...])

build_encoder_branch(inputs, hidden_dim, z_dim)

Defines a single hidden layer encoder that outputs two encodings: mu and log_var.

build_latent_space_classifier(inputs, ...[, ...])

A simple classifier that predicts from the latent space representation.

build_prop_estimator(inputs, num_classes[, ...])

Defines a classifier network that estimates cell type proportion from the input.

wrapped_build_decoder_branch(...[, ...])

Factory wrapper around build_decoder_branch.

wrapped_build_encoder_branch(input_shape, ...)

Factory wrapper around build_encoder_branch.

wrapped_build_latent_space_classifier_branch(...)

Factory wrapper around build_latent_space_classifier.

wrapped_build_prop_estimator_branch(...[, ...])

Factory wrapper around build_prop_estimator.

Classes

Model(*args, **kwargs)

A model grouping layers into an object with training/inference features.