aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/main/java/com/yahoo/search/ranking/GlobalPhaseSetup.java
diff options
context:
space:
mode:
Diffstat (limited to 'container-search/src/main/java/com/yahoo/search/ranking/GlobalPhaseSetup.java')
-rw-r--r--container-search/src/main/java/com/yahoo/search/ranking/GlobalPhaseSetup.java16
1 files changed, 7 insertions, 9 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/ranking/GlobalPhaseSetup.java b/container-search/src/main/java/com/yahoo/search/ranking/GlobalPhaseSetup.java
index 31a676e4c8e..e5cd09d3a18 100644
--- a/container-search/src/main/java/com/yahoo/search/ranking/GlobalPhaseSetup.java
+++ b/container-search/src/main/java/com/yahoo/search/ranking/GlobalPhaseSetup.java
@@ -3,15 +3,10 @@ package com.yahoo.search.ranking;
import ai.vespa.models.evaluation.FunctionEvaluator;
+import com.yahoo.tensor.Tensor;
import com.yahoo.vespa.config.search.RankProfilesConfig;
-import java.util.ArrayList;
-import java.util.Collection;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
-import java.util.Map;
-import java.util.HashMap;
+import java.util.*;
import java.util.function.Supplier;
class GlobalPhaseSetup {
@@ -20,16 +15,19 @@ class GlobalPhaseSetup {
final int rerankCount;
final Collection<String> matchFeaturesToHide;
final List<NormalizerSetup> normalizers;
+ final Map<String, Tensor> defaultValues;
GlobalPhaseSetup(FunEvalSpec globalPhaseEvalSpec,
final int rerankCount,
Collection<String> matchFeaturesToHide,
- List<NormalizerSetup> normalizers)
+ List<NormalizerSetup> normalizers,
+ Map<String, Tensor> defaultValues)
{
this.globalPhaseEvalSpec = globalPhaseEvalSpec;
this.rerankCount = rerankCount;
this.matchFeaturesToHide = matchFeaturesToHide;
this.normalizers = normalizers;
+ this.defaultValues = defaultValues;
}
static GlobalPhaseSetup maybeMakeSetup(RankProfilesConfig.Rankprofile rp, RankProfilesEvaluator modelEvaluator) {
@@ -106,7 +104,7 @@ class GlobalPhaseSetup {
}
Supplier<Evaluator> supplier = SimpleEvaluator.wrap(functionEvaluatorSource);
var gfun = new FunEvalSpec(supplier, fromQuery, fromMF);
- return new GlobalPhaseSetup(gfun, rerankCount, namesToHide, normalizers);
+ return new GlobalPhaseSetup(gfun, rerankCount, namesToHide, normalizers, Collections.emptyMap());
}
return null;
}