SURE-tools 4.0.0__py3-none-any.whl → 4.0.1__py3-none-any.whl

SURE/SURE_vae.py

@@ -294,7 +294,7 @@ class SUREVAE(nn.Module):
             )
             
         self.decoder_log_mu = MLP(
-                [self.latent_dim+self.latent_dim+self.latent_dim] + self.decoder_hidden_layers + [self.input_dim],
+                [self.latent_dim] + self.decoder_hidden_layers + [self.input_dim],
                 activation=activate_fct,
                 output_activation=None,
                 post_layer_fct=post_layer_fct,
@@ -436,7 +436,7 @@ class SUREVAE(nn.Module):
             else:
                 zfs = torch.zeros_like(zs)
 
-            log_mu = self.decoder_log_mu([zs+zcs,zps,zfs])
+            log_mu = self.decoder_log_mu(zs+zcs+zps+zfs)
             if self.loss_func in ['bernoulli']:
                 log_theta = log_mu
             elif self.loss_func in ['negbinomial']:
@@ -699,7 +699,7 @@ class SUREVAE(nn.Module):
                     
                 zfs = torch.zeros_like(z_basal)
                 
-                log_mu = self.decoder_log_mu([z_basal+zcs, zps, zfs])
+                log_mu = self.decoder_log_mu(z_basal+zcs+zps+zfs)
                 if self.loss_func == 'bernoulli':
                     counts = dist.Bernoulli(logits=log_mu).to_event(1).mean
                 else:

{sure_tools-4.0.0.dist-info → sure_tools-4.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 4.0.0
+Version: 4.0.1
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

{sure_tools-4.0.0.dist-info → sure_tools-4.0.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 SURE/SURE.py,sha256=8E39np6zhLbT1cp1xYOg5xLwzFHBupKIm1ydLxNKtqM,16654
 SURE/SUREMO.py,sha256=hN0G0ZEBNQdmj0gGlBIy1wjKGKqAMUemDjQeICzvNUY,47644
 SURE/SURE_nsf.py,sha256=VR6YgEiIfu7mRH0XLbovfjd2X3WFuom2J4_AgHrq2dM,49040
-SURE/SURE_vae.py,sha256=fexWxK80CTLtA38Y9kePU4SndmmEEwssJt8kWpttI08,45888
+SURE/SURE_vae.py,sha256=b3xluu49kmuDlcpZnTAJc92SvQI8MPDN9ntBfnKTJOQ,45850
 SURE/SURE_vanilla.py,sha256=I1RHHCpzk8ml1vMdH_gITOzAFXrHYUA7IAwbVSoxmBo,27327
 SURE/__init__.py,sha256=eJN0vlGblWir1JHaoiQqbQHzS_C1PNYcA_ls7UviTqc,444
 SURE/atac/__init__.py,sha256=3smP8IKHfwNCd1G_sZH3pKHXuLkLpFuLtjUTUSy7_As,34
@@ -15,8 +15,8 @@ SURE/utils/custom_mlp.py,sha256=XvviNUYeaZ5D1evqETyWLxgtL56mmFQR_pxsYpKw0yY,1170
 SURE/utils/label.py,sha256=joKO1mSkjZXeLvSeC7GluQk4-_qgGgPqlwWixdcbKMQ,4648
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-4.0.0.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-4.0.0.dist-info/METADATA,sha256=Yb-A4B_KUItpuJKnG0XKBKrVCkGI1xp0gHY5w_itVUI,1661
-sure_tools-4.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-4.0.0.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-4.0.0.dist-info/RECORD,,
+sure_tools-4.0.1.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-4.0.1.dist-info/METADATA,sha256=m4l4x8KnzdyfZquikfwrSqZXegaeRl6X9plr-fcuGrc,1661
+sure_tools-4.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-4.0.1.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-4.0.1.dist-info/RECORD,,