SURE-tools 4.0.4__tar.gz → 4.0.6__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.4 → sure_tools-4.0.6}/PKG-INFO +1 -1
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/SURE_nsf.py +1 -1
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/SURE_vae.py +2 -2
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-4.0.4 → sure_tools-4.0.6}/setup.py +1 -1
- {sure_tools-4.0.4 → sure_tools-4.0.6}/LICENSE +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/README.md +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/SURE.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/SUREMO.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/SURE_vanilla.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/__init__.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/atac/__init__.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/atac/utils.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/dist/__init__.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/dist/negbinomial.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/graph/__init__.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/graph/graph_utils.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/utils/__init__.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/utils/custom_mlp.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/utils/label.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/utils/queue.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE/utils/utils.py +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-4.0.4 → sure_tools-4.0.6}/setup.cfg +0 -0
|
@@ -424,7 +424,7 @@ class SURENF(nn.Module):
|
|
|
424
424
|
zps = self.perturb_effect([ps, zs+zcs])
|
|
425
425
|
else:
|
|
426
426
|
zps = torch.zeros_like(zs)
|
|
427
|
-
if (self.covariate_sizes>0) and (fs is not None):
|
|
427
|
+
if (np.sum(self.covariate_sizes)>0) and (fs is not None):
|
|
428
428
|
zfs = torch.zeros_like(zs)
|
|
429
429
|
shift = 0
|
|
430
430
|
for i, covariate_size in enumerate(self.covariate_sizes):
|
|
@@ -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,zps,zfs)
|
|
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,zps,zfs)
|
|
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
|