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

bugfix in predict_model for hdf5 larva_dataset files

parent dc04b9af
No related branches found
No related tags found
2 merge requests!11Set of commits to be tagged v0.18,!10Set of commits to be tagged v0.17
This commit is part of merge request !10. Comments created here will be created in the context of that merge request.
...@@ -150,7 +150,7 @@ def predict_individual_data_files(backend, model, input_files_and_labels): ...@@ -150,7 +150,7 @@ def predict_individual_data_files(backend, model, input_files_and_labels):
def predict_larva_dataset(backend, model, file, subset="validation", subsets=(.8, .2, 0)): def predict_larva_dataset(backend, model, file, subset="validation", subsets=(.8, .2, 0)):
from taggingbackends.data.dataset import LarvaDataset from taggingbackends.data.dataset import LarvaDataset
dataset = LarvaDataset(file, new_generator(), subsets) dataset = LarvaDataset(file, new_generator(), subsets)
return model.predict(dataset, subset) return model.predict((dataset, subset))
def _zip(xs, ys): def _zip(xs, ys):
# prevent issues similar to #2 # prevent issues similar to #2
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment