Skip to content
Snippets Groups Projects

Added distance prediction and scoring

Merged Yannick DAYER requested to merge algorithm_fix into master
1 file
+ 11
4
Compare changes
  • Side-by-side
  • Inline
@@ -581,9 +581,12 @@ class AlgorithmAdaptor:
@@ -581,9 +581,12 @@ class AlgorithmAdaptor:
def enroll(self, k):
def enroll(self, k):
self.load()
self.load()
return self.model.enroll(
if self.model.requires_projector_training:
[self.model.project(s.data) for s in k.samples]
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):
def write_enrolled(self, k, path):
self.model.write_model(k, path)
self.model.write_model(k, path)
@@ -652,7 +655,11 @@ class AlgorithmAdaptor:
@@ -652,7 +655,11 @@ class AlgorithmAdaptor:
retval = []
retval = []
for p in probes:
for p in probes:
data = [model.project(s.data) for s in p.samples]
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)):
for subprobe_id, (s, parent) in enumerate(zip(data, p.samples)):
# each sub-probe in the probe needs to be checked
# each sub-probe in the probe needs to be checked
subprobe_scores = []
subprobe_scores = []
Loading