SURE-tools 4.0.3__py3-none-any.whl → 4.0.5__py3-none-any.whl

SURE/SURE_vae.py

@@ -297,7 +297,7 @@ class SUREVAE(nn.Module):
                 )
             
         self.decoder_log_mu = MLP(
-                [self.latent_dim] + self.decoder_hidden_layers + [self.input_dim],
+                [self.latent_dim+self.latent_dim+self.latent_dim] + self.decoder_hidden_layers + [self.input_dim],
                 activation=activate_fct,
                 output_activation=None,
                 post_layer_fct=post_layer_fct,
@@ -444,7 +444,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']:
@@ -707,7 +707,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.3.dist-info → sure_tools-4.0.5.dist-info}/METADATA

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

{sure_tools-4.0.3.dist-info → sure_tools-4.0.5.dist-info}/RECORD

@@ -1,7 +1,7 @@
 SURE/SURE.py,sha256=YcJ6Lxt4ziQXZ5Yk02Ss-fJRl_M1Z0QghLY_yeHt5Pc,16659
 SURE/SUREMO.py,sha256=hN0G0ZEBNQdmj0gGlBIy1wjKGKqAMUemDjQeICzvNUY,47644
 SURE/SURE_nsf.py,sha256=uKpau3nDdlX58lLzPzecZULKobyOYB7xKtcYo6vwbak,49541
-SURE/SURE_vae.py,sha256=X4aNb6Wv9Xn-w1Xc-wUuio6YsxX0dIwdJqsWG44hCa0,46375
+SURE/SURE_vae.py,sha256=VhJk9K_RA6IzaZ8WpP8T3j1c5I0_o7x6VZzyP_QBseI,46411
 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.3.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-4.0.3.dist-info/METADATA,sha256=DlTyOxPOevVRibU7zjiIorXK-Te5I5uMI3zec3H94zw,1661
-sure_tools-4.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-4.0.3.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-4.0.3.dist-info/RECORD,,
+sure_tools-4.0.5.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-4.0.5.dist-info/METADATA,sha256=aB-j0mgfSBnmu6E-OM6a6UcXpckqbmbCM5LALP-o5NM,1661
+sure_tools-4.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-4.0.5.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-4.0.5.dist-info/RECORD,,