SURE-tools 4.0.6__tar.gz → 4.0.8__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.6 → sure_tools-4.0.8}/PKG-INFO +1 -1
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/SURE_nsf.py +3 -3
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/SURE_vae.py +3 -3
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-4.0.6 → sure_tools-4.0.8}/setup.py +1 -1
- {sure_tools-4.0.6 → sure_tools-4.0.8}/LICENSE +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/README.md +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/SURE.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/SUREMO.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/SURE_vanilla.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/__init__.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/atac/__init__.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/atac/utils.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/dist/__init__.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/dist/negbinomial.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/graph/__init__.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/graph/graph_utils.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/utils/__init__.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/utils/custom_mlp.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/utils/label.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/utils/queue.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE/utils/utils.py +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-4.0.6 → sure_tools-4.0.8}/setup.cfg +0 -0
|
@@ -287,7 +287,7 @@ class SURENF(nn.Module):
|
|
|
287
287
|
)
|
|
288
288
|
|
|
289
289
|
self.decoder_log_mu = MLP(
|
|
290
|
-
[self.latent_dim
|
|
290
|
+
[self.latent_dim] + self.decoder_hidden_layers + [self.input_dim],
|
|
291
291
|
activation=activate_fct,
|
|
292
292
|
output_activation=None,
|
|
293
293
|
post_layer_fct=post_layer_fct,
|
|
@@ -434,7 +434,7 @@ class SURENF(nn.Module):
|
|
|
434
434
|
else:
|
|
435
435
|
zfs = torch.zeros_like(zs)
|
|
436
436
|
|
|
437
|
-
log_mu = self.decoder_log_mu(
|
|
437
|
+
log_mu = self.decoder_log_mu(zs+zcs+zps+zfs)
|
|
438
438
|
if self.loss_func in ['bernoulli']:
|
|
439
439
|
log_theta = log_mu
|
|
440
440
|
elif self.loss_func in ['negbinomial']:
|
|
@@ -692,7 +692,7 @@ class SURENF(nn.Module):
|
|
|
692
692
|
|
|
693
693
|
zfs = torch.zeros_like(z_basal)
|
|
694
694
|
|
|
695
|
-
log_mu = self.decoder_log_mu(
|
|
695
|
+
log_mu = self.decoder_log_mu(z_basal+zcs+zps+zfs)
|
|
696
696
|
if self.loss_func == 'bernoulli':
|
|
697
697
|
counts = dist.Bernoulli(logits=log_mu).to_event(1).mean
|
|
698
698
|
else:
|
|
@@ -297,7 +297,7 @@ class SUREVAE(nn.Module):
|
|
|
297
297
|
)
|
|
298
298
|
|
|
299
299
|
self.decoder_log_mu = MLP(
|
|
300
|
-
[self.latent_dim
|
|
300
|
+
[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(
|
|
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(
|
|
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
|