cpa.CPAModule.inference

CPAModule.inference(x, perts, perts_doses, covars_dict, mixup_lambda=1.0, n_samples=1, covars_to_add=None)

Run the recognition model.

In the case of variational inference, this function will perform steps related to computing variational distribution parameters. In a VAE, this will involve running data through encoder networks.

This function should return a dictionary with str keys and Tensor values.