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
Branches
No related tags found
2 merge requests!11Set of commits to be tagged v0.18,!10Set of commits to be tagged v0.17
...@@ -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