sfaira.estimators.EstimatorKerasEmbedding.predict_embedding

EstimatorKerasEmbedding.predict_embedding(batch_size: int = 128, variational: bool = False)

return the prediction in the latent space (z_mean for variational models)

Params variational

Whether toreturn the prediction of z, z_mean, z_log_var in the variational latent space.

Returns

latent space