Using max rank parameter.

Signed-off-by: jzonthemtn <[email protected]>
o19s · Dec 11, 2024 · cc951e6 · cc951e6
1 parent 5c09b61
commit cc951e6
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/...on-plugin/src/main/java/org/opensearch/eval/judgments/clickmodel/coec/CoecClickModel.java b/...on-plugin/src/main/java/org/opensearch/eval/judgments/clickmodel/coec/CoecClickModel.java
@@ -111,9 +111,6 @@ public String calculateCoec(final Map<Integer, Double> rankAggregatedClickThroug
         // Format: query_id, query, document, judgment
         final Collection<Judgment> judgments = new LinkedList<>();
 
-        // Up to Rank R
-        final int maxRank = 20;
-
         LOGGER.info("Count of queries: {}", clickthroughRates.size());
 
         for(final String userQuery : clickthroughRates.keySet()) {
@@ -127,7 +124,7 @@ public String calculateCoec(final Map<Integer, Double> rankAggregatedClickThroug
 
                 double denominatorSum = 0;
 
-                for(int rank = 0; rank < maxRank; rank++) {
+                for(int rank = 0; rank < parameters.getMaxRank(); rank++) {
 
                     // The document's mean CTR at the rank.
                     final double meanCtrAtRank = rankAggregatedClickThrough.getOrDefault(rank, 0.0);