buddi.models.components.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.

build_semi_supervised_decoder(...[, ...])

Defines a decoder network that takes in both supervised and unsupervised inputs.

Classes

Concatenate(*args, **kwargs)

Concatenates a list of inputs.

Dense(*args, **kwargs)

Just your regular densely-connected NN layer.

Model(*args, **kwargs)

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

ReLU(*args, **kwargs)

Rectified Linear Unit activation function layer.