|
17 | 17 | import org.elasticsearch.index.rankeval.ExpectedReciprocalRank;
|
18 | 18 | import org.elasticsearch.index.rankeval.MeanReciprocalRank;
|
19 | 19 | import org.elasticsearch.index.rankeval.PrecisionAtK;
|
20 |
| -import org.elasticsearch.index.rankeval.RecallAtK; |
21 | 20 | import org.elasticsearch.index.rankeval.RankEvalRequest;
|
22 | 21 | import org.elasticsearch.index.rankeval.RankEvalResponse;
|
23 | 22 | import org.elasticsearch.index.rankeval.RankEvalSpec;
|
24 | 23 | import org.elasticsearch.index.rankeval.RatedDocument;
|
25 | 24 | import org.elasticsearch.index.rankeval.RatedRequest;
|
26 | 25 | import org.elasticsearch.index.rankeval.RatedSearchHit;
|
| 26 | +import org.elasticsearch.index.rankeval.RecallAtK; |
27 | 27 | import org.elasticsearch.search.builder.SearchSourceBuilder;
|
28 | 28 | import org.junit.Before;
|
29 | 29 |
|
@@ -68,7 +68,7 @@ public void testRankEvalRequest() throws IOException {
|
68 | 68 | RankEvalResponse response = execute(rankEvalRequest, highLevelClient()::rankEval, highLevelClient()::rankEvalAsync);
|
69 | 69 | // the expected Prec@ for the first query is 5/7 and the expected Prec@ for the second is 1/7, divided by 2 to get the average
|
70 | 70 | double expectedPrecision = (1.0 / 7.0 + 5.0 / 7.0) / 2.0;
|
71 |
| - assertEquals(expectedPrecision, response.getMetricScore(), Double.MIN_VALUE); |
| 71 | + assertEquals(expectedPrecision, response.getMetricScore(), 0.00000000001); |
72 | 72 | Map<String, EvalQueryQuality> partialResults = response.getPartialResults();
|
73 | 73 | assertEquals(2, partialResults.size());
|
74 | 74 | EvalQueryQuality amsterdamQueryQuality = partialResults.get("amsterdam_query");
|
@@ -129,7 +129,7 @@ public void testMetrics() throws IOException {
|
129 | 129 |
|
130 | 130 | RankEvalRequest rankEvalRequest = new RankEvalRequest(spec, new String[] { "index", "index2" });
|
131 | 131 | RankEvalResponse response = execute(rankEvalRequest, highLevelClient()::rankEval, highLevelClient()::rankEvalAsync);
|
132 |
| - assertEquals(expectedScores[i], response.getMetricScore(), Double.MIN_VALUE); |
| 132 | + assertEquals(expectedScores[i], response.getMetricScore(), 0.00000000001); |
133 | 133 | i++;
|
134 | 134 | }
|
135 | 135 | }
|
|
0 commit comments