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
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