poli.objective_repository.gfp_cbas.cbas_wrapper.CBASVAEWrapper

poli.objective_repository.gfp_cbas.cbas_wrapper.CBASVAEWrapper#

class poli.objective_repository.gfp_cbas.cbas_wrapper.CBASVAEWrapper(AA: int, L: int)#
__init__(AA: int, L: int)#

Methods

__init__(AA, L)

decode(z[, grad])

Returns an NxLxAA tensor.

decode_samples_from_encoder(x, num_samples)

For a single one hot encoded LxAA vector this method samples around the encoded sequence and returns the decoded sequences.

encode(x[, grad])

Turns a one hot encoded LxAA vector into a continuous lower dimensional vector.

encode_sequences(seqs)

get_latent_dimensionality()

get_max_llh_sequence(proba)

get_wildtype_distribution()

Returns a distribution to represent the wild-type.

transform_sequences_to_atoms(x)

Turns a sequence into an approximately Dirac distribution around the sequence.