Skip to content

Commit 00169b2

Browse files
committed
Node feature added
1 parent 5716501 commit 00169b2

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

x-pack/plugin/rank-rrf/src/main/java/org/elasticsearch/xpack/rank/RankRRFFeatures.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import java.util.Set;
1414

1515
import static org.elasticsearch.search.retriever.CompoundRetrieverBuilder.INNER_RETRIEVERS_FILTER_SUPPORT;
16+
import static org.elasticsearch.xpack.rank.linear.L2ScoreNormalizer.LINEAR_RETRIEVER_L2_NORM;
1617
import static org.elasticsearch.xpack.rank.linear.MinMaxScoreNormalizer.LINEAR_RETRIEVER_MINMAX_SINGLE_DOC_FIX;
1718

1819
public class RankRRFFeatures implements FeatureSpecification {
@@ -26,6 +27,6 @@ public Set<NodeFeature> getFeatures() {
2627

2728
@Override
2829
public Set<NodeFeature> getTestFeatures() {
29-
return Set.of(INNER_RETRIEVERS_FILTER_SUPPORT, LINEAR_RETRIEVER_MINMAX_SINGLE_DOC_FIX);
30+
return Set.of(INNER_RETRIEVERS_FILTER_SUPPORT, LINEAR_RETRIEVER_MINMAX_SINGLE_DOC_FIX, LINEAR_RETRIEVER_L2_NORM);
3031
}
3132
}

x-pack/plugin/rank-rrf/src/main/java/org/elasticsearch/xpack/rank/linear/L2ScoreNormalizer.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
package org.elasticsearch.xpack.rank.linear;
1010

1111
import org.apache.lucene.search.ScoreDoc;
12+
import org.elasticsearch.features.NodeFeature;
1213

1314
/**
1415
* A score normalizer that applies L2 normalization to a set of scores.
@@ -25,6 +26,8 @@ public class L2ScoreNormalizer extends ScoreNormalizer {
2526

2627
private static final float EPSILON = 1e-6f;
2728

29+
public static final NodeFeature LINEAR_RETRIEVER_L2_NORM = new NodeFeature("linear_retriever.l2_norm");
30+
2831
public L2ScoreNormalizer() {}
2932

3033
@Override

0 commit comments

Comments
 (0)