blob: 548576e3a1581ac6b834b2e5a59bea4af27e4109 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.ranking;
import ai.vespa.models.evaluation.FunctionEvaluator;
import com.yahoo.search.result.FeatureData;
import com.yahoo.search.result.Hit;
import com.yahoo.tensor.Tensor;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.function.Supplier;
class SimpleEvaluator implements Evaluator {
private final FunctionEvaluator evaluator;
static Supplier<Evaluator> wrap(Supplier<FunctionEvaluator> supplier) {
return () -> new SimpleEvaluator(supplier.get());
}
SimpleEvaluator(FunctionEvaluator prototype) {
this.evaluator = prototype;
}
@Override
public Evaluator bind(String name, Tensor value) {
evaluator.bind(name, value);
return this;
}
@Override
public double evaluateScore() {
return evaluator.evaluate().asDouble();
}
@Override
public String toString() {
var buf = new StringBuilder();
buf.append("SimpleEvaluator(");
buf.append(evaluator.function().toString());
buf.append(")[");
for (String arg : evaluator.function().arguments()) {
buf.append("{").append(arg).append("}");
}
buf.append("]");
return buf.toString();
}
}
|