SURE-tools 2.2.28__tar.gz → 2.3.1__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.
Potentially problematic release.
This version of SURE-tools might be problematic. Click here for more details.
- {sure_tools-2.2.28 → sure_tools-2.3.1}/PKG-INFO +1 -1
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/DensityFlow.py +4 -4
- sure_tools-2.3.1/SURE/PerturbE.py +1293 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/__init__.py +2 -1
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/utils/custom_mlp.py +31 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE_tools.egg-info/SOURCES.txt +1 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/setup.py +1 -1
- {sure_tools-2.2.28 → sure_tools-2.3.1}/LICENSE +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/README.md +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/SURE.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/assembly/__init__.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/assembly/assembly.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/assembly/atlas.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/atac/__init__.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/atac/utils.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/codebook/__init__.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/codebook/codebook.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/flow/__init__.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/flow/flow_stats.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/flow/plot_quiver.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/perturb/__init__.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/perturb/perturb.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/utils/__init__.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/utils/queue.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE/utils/utils.py +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE_tools.egg-info/entry_points.txt +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-2.2.28 → sure_tools-2.3.1}/setup.cfg +0 -0
|
@@ -1340,7 +1340,7 @@ def main():
|
|
|
1340
1340
|
cell_factor_size = 0 if us is None else us.shape[1]
|
|
1341
1341
|
|
|
1342
1342
|
###########################################
|
|
1343
|
-
|
|
1343
|
+
df = DensityFlow(
|
|
1344
1344
|
input_size=input_size,
|
|
1345
1345
|
cell_factor_size=cell_factor_size,
|
|
1346
1346
|
inverse_dispersion=args.inverse_dispersion,
|
|
@@ -1359,7 +1359,7 @@ def main():
|
|
|
1359
1359
|
dtype=dtype,
|
|
1360
1360
|
)
|
|
1361
1361
|
|
|
1362
|
-
|
|
1362
|
+
df.fit(xs, us=us,
|
|
1363
1363
|
num_epochs=args.num_epochs,
|
|
1364
1364
|
learning_rate=args.learning_rate,
|
|
1365
1365
|
batch_size=args.batch_size,
|
|
@@ -1371,9 +1371,9 @@ def main():
|
|
|
1371
1371
|
|
|
1372
1372
|
if args.save_model is not None:
|
|
1373
1373
|
if args.save_model.endswith('gz'):
|
|
1374
|
-
DensityFlow.save_model(
|
|
1374
|
+
DensityFlow.save_model(df, args.save_model, compression=True)
|
|
1375
1375
|
else:
|
|
1376
|
-
DensityFlow.save_model(
|
|
1376
|
+
DensityFlow.save_model(df, args.save_model)
|
|
1377
1377
|
|
|
1378
1378
|
|
|
1379
1379
|
|