Skip to content
Snippets Groups Projects
Commit 21db7e8c authored by François  LAURENT's avatar François LAURENT
Browse files

balancing_strategy management moved to taggingbackends

parent a02dc0f4
No related branches found
No related tags found
1 merge request!3Set of commits to be tagged v0.16.1
...@@ -14,7 +14,7 @@ maggotuba-core = {git = "https://gitlab.pasteur.fr/nyx/MaggotUBA-core", tag = "v ...@@ -14,7 +14,7 @@ maggotuba-core = {git = "https://gitlab.pasteur.fr/nyx/MaggotUBA-core", tag = "v
torch = "^1.11.0" torch = "^1.11.0"
numpy = "^1.19.3" numpy = "^1.19.3"
protobuf = "3.9.2" protobuf = "3.9.2"
taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.14.1"} taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.15"}
[build-system] [build-system]
requires = ["poetry-core>=1.0.0"] requires = ["poetry-core>=1.0.0"]
......
...@@ -59,10 +59,10 @@ def make_dataset(backend, labels_expected=False, trxmat_only=False, ...@@ -59,10 +59,10 @@ def make_dataset(backend, labels_expected=False, trxmat_only=False,
print("generating a larva_dataset file...") print("generating a larva_dataset file...")
# generate a larva_dataset_*.hdf5 file in data/interim/{instance}/ # generate a larva_dataset_*.hdf5 file in data/interim/{instance}/
balance = isinstance(balancing_strategy, str) and balancing_strategy.lower() == 'maggotuba'
out = backend.generate_dataset(backend.raw_data_dir(), out = backend.generate_dataset(backend.raw_data_dir(),
past_future_extensions=False, past_future_extensions=False,
balance=balance, **kwargs) balancing_strategy=balancing_strategy,
**kwargs)
print(f"larva_dataset file generated: {out}") print(f"larva_dataset file generated: {out}")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment