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

bugfix in predict_model for hdf5 larva_dataset files

parent d5f69f19
No related branches found
No related tags found
1 merge request!11Set of commits to be tagged v0.18
......@@ -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)):
from taggingbackends.data.dataset import LarvaDataset
dataset = LarvaDataset(file, new_generator(), subsets)
return model.predict(dataset, subset)
return model.predict((dataset, subset))
def _zip(xs, ys):
# 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