scvi.external.gimvi.JVAE.sample_rate

JVAE.sample_rate(x, mode, batch_index, y=None, deterministic=False, decode_mode=None)[source]

Returns the tensor of scaled frequencies of expression.

Parameters
x : TensorTensor

tensor of values with shape (batch_size, n_input) or (batch_size, n_input_fish) depending on the mode

y : Tensor | NoneOptional[Tensor] (default: None)

tensor of cell-types labels with shape (batch_size, n_labels)

mode : intint

int encode mode (which input head to use in the model)

batch_index : TensorTensor

array that indicates which batch the cells belong to with shape batch_size

deterministic : boolbool (default: False)

bool - whether to sample or not

decode_mode : int | NoneOptional[int] (default: None)

int use to a decode mode different from encoding mode

Return type

TensorTensor

Returns

type tensor of means of the scaled frequencies