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

bugfix with path property

parent 20fd7d2a
No related branches found
No related tags found
1 merge request!10Set of commits to be tagged v0.17
Pipeline #119487 passed
......@@ -10,6 +10,7 @@ Torch-like dataset class for *larva_dataset hdf5* files.
class LarvaDataset:
def __init__(self, dataset=None, generator=None, subsets=(.8, .1, .1),
balancing_strategy=None, class_weights=None):
self._path = None
self.generator = generator
self._full_set = dataset
self.subset_shares = subsets
......@@ -26,7 +27,6 @@ class LarvaDataset:
self.weight_classes = isinstance(balancing_strategy, str) and (balancing_strategy.lower() == 'auto')
else:
self.class_weights = class_weights
self._path = None
"""
*h5py.File*: *larva_dataset hdf5* file handler.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment