container-search/src/main/java/com/yahoo/search/ranking/SimpleEvaluator.java


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
51

// Copyright Yahoo. 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;

class SimpleEvaluator implements Evaluator {

    private final FunctionEvaluator evaluator;
    private final Set<String> neededInputs;
    
    public SimpleEvaluator(FunctionEvaluator prototype) {
        this.evaluator = prototype;
        this.neededInputs = new HashSet<String>(prototype.function().arguments());
    }

    @Override
    public Collection<String> needInputs() { return List.copyOf(neededInputs); }

    @Override
    public SimpleEvaluator bind(String name, Tensor value) {
        if (value != null) evaluator.bind(name, value);
        neededInputs.remove(name);
        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 : neededInputs) {
            buf.append("{").append(arg).append("}");
        }
        buf.append("]");
        return buf.toString();
    }
}