SURE-tools 2.2.20__tar.gz → 2.2.27__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-2.2.20 → sure_tools-2.2.27}/PKG-INFO +1 -1
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/DensityFlow.py +2 -2
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/utils/custom_mlp.py +9 -2
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-2.2.20 → sure_tools-2.2.27}/setup.py +1 -1
- {sure_tools-2.2.20 → sure_tools-2.2.27}/LICENSE +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/README.md +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/SURE.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/__init__.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/assembly/__init__.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/assembly/assembly.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/assembly/atlas.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/atac/__init__.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/atac/utils.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/codebook/__init__.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/codebook/codebook.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/flow/__init__.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/flow/flow_stats.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/flow/plot_quiver.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/perturb/__init__.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/perturb/perturb.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/utils/__init__.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/utils/queue.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE/utils/utils.py +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE_tools.egg-info/entry_points.txt +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-2.2.20 → sure_tools-2.2.27}/setup.cfg +0 -0
|
@@ -854,12 +854,12 @@ class DensityFlow(nn.Module):
|
|
|
854
854
|
us_i = us[:,pert_idx].reshape(-1,1)
|
|
855
855
|
|
|
856
856
|
# factor effect of xs
|
|
857
|
-
dzs0 = self.get_cell_response(zs,
|
|
857
|
+
dzs0 = self.get_cell_response(zs, perturb_idx=pert_idx, perturb_us=us_i)
|
|
858
858
|
|
|
859
859
|
# perturbation effect
|
|
860
860
|
ps = np.ones_like(us_i)
|
|
861
861
|
if np.sum(np.abs(ps-us_i))>=1:
|
|
862
|
-
dzs = self.get_cell_response(zs,
|
|
862
|
+
dzs = self.get_cell_response(zs, perturb_idx=pert_idx, perturb_us=ps)
|
|
863
863
|
zs = zs + dzs0 + dzs
|
|
864
864
|
else:
|
|
865
865
|
zs = zs + dzs0
|
|
@@ -237,12 +237,19 @@ class ZeroBiasMLP(nn.Module):
|
|
|
237
237
|
|
|
238
238
|
# pass through our sequential for the output!
|
|
239
239
|
def forward(self, x):
|
|
240
|
+
print(f'type: {type(x)}; shape: {x.shape}')
|
|
240
241
|
y = self.mlp(x)
|
|
241
242
|
mask = torch.zeros_like(y)
|
|
242
243
|
if len(y.shape)==2:
|
|
243
|
-
|
|
244
|
+
if type(x)==list:
|
|
245
|
+
mask[x[1][:,0]>0,:] = 1
|
|
246
|
+
else:
|
|
247
|
+
mask[x[:,0]>0,:] = 1
|
|
244
248
|
elif len(y.shape)==3:
|
|
245
|
-
|
|
249
|
+
if type(x)==list:
|
|
250
|
+
mask[:,x[1][:,0]>0,:] = 1
|
|
251
|
+
else:
|
|
252
|
+
mask[:,x[:,0]>0,:] = 1
|
|
246
253
|
return y*mask
|
|
247
254
|
|
|
248
255
|
|
|
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
|
|
File without changes
|
|
File without changes
|