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

bugfix in predict_model for hdf5 larva_dataset files

parent 7bd32604
Branches
Tags
1 merge request!11Set of commits to be tagged v0.18
...@@ -152,7 +152,7 @@ def predict_individual_data_files(backend, model, input_files_and_labels): ...@@ -152,7 +152,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