Skip to content
Snippets Groups Projects
Commit bac72e8a authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Merge branch 'algorithm_fix' into 'master'

Added distance prediction and scoring

See merge request !2
parents 72ff8227 c22c0051
No related branches found
No related tags found
1 merge request!2Added distance prediction and scoring
Pipeline #37565 passed
......@@ -581,9 +581,12 @@ class AlgorithmAdaptor:
def enroll(self, k):
self.load()
if self.model.requires_projector_training:
return self.model.enroll(
[self.model.project(s.data) for s in k.samples]
)
else:
return [s.data for s in k.samples]
def write_enrolled(self, k, path):
self.model.write_model(k, path)
......@@ -652,7 +655,11 @@ class AlgorithmAdaptor:
retval = []
for p in probes:
if model.requires_projector_training:
data = [model.project(s.data) for s in p.samples]
else:
data = [s.data for s in p.samples]
for subprobe_id, (s, parent) in enumerate(zip(data, p.samples)):
# each sub-probe in the probe needs to be checked
subprobe_scores = []
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment