MultiEncoder¶

class scvi.models.modules.MultiEncoder(n_heads, n_input_list, n_output, n_hidden=128, n_layers_individual=1, n_layers_shared=2, n_cat_list=None, dropout_rate=0.1)[source]¶

Bases: torch.nn.modules.module.Module

Methods Summary

forward(x, head_id, *cat_list)

Methods Documentation

forward(x, head_id, *cat_list)[source]¶