Commit ea3a1441 authored by Manuel Günther's avatar Manuel Günther
Browse files

Used the correct score averaging function for models.

parent a43b31fd
Pipeline #56922 passed with stage
in 15 minutes and 2 seconds
......@@ -216,9 +216,9 @@ class Algorithm (object):
"""
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):
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:
raise ValueError("The model does not have the desired format (list, array, ...)")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment