Skip to content
Snippets Groups Projects
Commit ea3a1441 authored by Manuel Günther's avatar Manuel Günther
Browse files

Used the correct score averaging function for models.

parent a43b31fd
No related branches found
No related tags found
1 merge request!270Resolve "In the Algorithm, the model_fusion_function is ignored"
Pipeline #56922 passed
...@@ -216,9 +216,9 @@ class Algorithm (object): ...@@ -216,9 +216,9 @@ class Algorithm (object):
""" """
if isinstance(models, list): if isinstance(models, list):
return [self.probe_fusion_function(self.score(model, probe)) for model in models] return [self.model_fusion_function(self.score(model, probe)) for model in models]
elif isinstance(models, numpy.ndarray): elif isinstance(models, numpy.ndarray):
return [self.probe_fusion_function(self.score(models[i,:], probe)) for i in range(models.shape[0])] return [self.model_fusion_function(self.score(models[i,:], probe)) for i in range(models.shape[0])]
else: else:
raise ValueError("The model does not have the desired format (list, array, ...)") raise ValueError("The model does not have the desired format (list, array, ...)")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment