SURE-tools 4.0.3__tar.gz → 4.0.4__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {sure_tools-4.0.3 → sure_tools-4.0.4}/PKG-INFO +1 -1
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/SURE_vae.py +3 -3
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-4.0.3 → sure_tools-4.0.4}/setup.py +1 -1
- {sure_tools-4.0.3 → sure_tools-4.0.4}/LICENSE +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/README.md +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/SURE.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/SUREMO.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/SURE_nsf.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/SURE_vanilla.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/__init__.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/atac/__init__.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/atac/utils.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/dist/__init__.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/dist/negbinomial.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/graph/__init__.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/graph/graph_utils.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/utils/__init__.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/utils/custom_mlp.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/utils/label.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/utils/queue.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE/utils/utils.py +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-4.0.3 → sure_tools-4.0.4}/setup.cfg +0 -0
|
@@ -297,7 +297,7 @@ class SUREVAE(nn.Module):
|
|
|
297
297
|
)
|
|
298
298
|
|
|
299
299
|
self.decoder_log_mu = MLP(
|
|
300
|
-
[self.latent_dim] + self.decoder_hidden_layers + [self.input_dim],
|
|
300
|
+
[self.latent_dim+self.latent_dim+self.latent_dim] + self.decoder_hidden_layers + [self.input_dim],
|
|
301
301
|
activation=activate_fct,
|
|
302
302
|
output_activation=None,
|
|
303
303
|
post_layer_fct=post_layer_fct,
|
|
@@ -444,7 +444,7 @@ class SUREVAE(nn.Module):
|
|
|
444
444
|
else:
|
|
445
445
|
zfs = torch.zeros_like(zs)
|
|
446
446
|
|
|
447
|
-
log_mu = self.decoder_log_mu(zs+zcs
|
|
447
|
+
log_mu = self.decoder_log_mu(zs+zcs,zps,zfs)
|
|
448
448
|
if self.loss_func in ['bernoulli']:
|
|
449
449
|
log_theta = log_mu
|
|
450
450
|
elif self.loss_func in ['negbinomial']:
|
|
@@ -707,7 +707,7 @@ class SUREVAE(nn.Module):
|
|
|
707
707
|
|
|
708
708
|
zfs = torch.zeros_like(z_basal)
|
|
709
709
|
|
|
710
|
-
log_mu = self.decoder_log_mu(z_basal+zcs
|
|
710
|
+
log_mu = self.decoder_log_mu(z_basal+zcs,zps,zfs)
|
|
711
711
|
if self.loss_func == 'bernoulli':
|
|
712
712
|
counts = dist.Bernoulli(logits=log_mu).to_event(1).mean
|
|
713
713
|
else:
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|