Move github review comments to TODOs

This commit is contained in:
Isabel Drost-Fromm 2016-07-21 15:44:47 +02:00
parent 2d57fd10b1
commit c8d3098d3e
8 changed files with 10 additions and 1 deletions

View File

@ -128,6 +128,7 @@ public class PrecisionAtN extends RankedListQualityMetric {
return new EvalQueryQuality(precision, unknownDocIds); return new EvalQueryQuality(precision, unknownDocIds);
} }
// TODO add abstraction that also works for other metrics
public enum Rating { public enum Rating {
IRRELEVANT, RELEVANT; IRRELEVANT, RELEVANT;
} }

View File

@ -38,6 +38,7 @@ import java.util.Map;
* Documents of unknown quality - i.e. those that haven't been supplied in the set of annotated documents but have been returned * Documents of unknown quality - i.e. those that haven't been supplied in the set of annotated documents but have been returned
* by the search are not taken into consideration when computing precision at n - they are ignored. * by the search are not taken into consideration when computing precision at n - they are ignored.
* *
* TODO get rid of either this or RankEvalResult
**/ **/
public class RankEvalResponse extends ActionResponse implements ToXContent { public class RankEvalResponse extends ActionResponse implements ToXContent {

View File

@ -32,6 +32,7 @@ import java.util.Map;
* for reference. In addition the averaged precision and the ids of all documents returned but not found annotated is returned. * for reference. In addition the averaged precision and the ids of all documents returned but not found annotated is returned.
* */ * */
// TODO do we need an extra class for this or it RankEvalResponse enough? // TODO do we need an extra class for this or it RankEvalResponse enough?
// TODO instead of just returning averages over complete results, think of other statistics, micro avg, macro avg, partial results
public class RankEvalResult implements Writeable { public class RankEvalResult implements Writeable {
/**ID of QA specification this result was generated for.*/ /**ID of QA specification this result was generated for.*/
private String specId; private String specId;

View File

@ -54,6 +54,7 @@ public abstract class RankedListQualityMetric implements NamedWriteable {
} }
String metricName = parser.currentName(); String metricName = parser.currentName();
// TODO maybe switch to using a plugable registry later?
switch (metricName) { switch (metricName) {
case PrecisionAtN.NAME: case PrecisionAtN.NAME:
rc = PrecisionAtN.fromXContent(parser, context); rc = PrecisionAtN.fromXContent(parser, context);

View File

@ -33,6 +33,7 @@ import java.io.IOException;
* */ * */
public class RatedDocument implements Writeable { public class RatedDocument implements Writeable {
// TODO augment with index name and type name
private final String docId; private final String docId;
private final int rating; private final int rating;

View File

@ -192,6 +192,7 @@ public class RestRankEvalAction extends BaseRestHandler {
QueryParseContext parseContext = new QueryParseContext(queryRegistry, parser, parseFieldMatcher); QueryParseContext parseContext = new QueryParseContext(queryRegistry, parser, parseFieldMatcher);
if (restContent != null) { if (restContent != null) {
parseRankEvalRequest(rankEvalRequest, request, parseRankEvalRequest(rankEvalRequest, request,
// TODO can we get rid of aggregators parsers and suggesters?
new RankEvalContext(parseFieldMatcher, parseContext, aggregators, suggesters)); new RankEvalContext(parseFieldMatcher, parseContext, aggregators, suggesters));
} }
} }

View File

@ -103,6 +103,7 @@ public class TransportRankEvalAction extends HandledTransportAction<RankEvalRequ
unknownDocs.put(spec.getSpecId(), intentQuality.getUnknownDocs()); unknownDocs.put(spec.getSpecId(), intentQuality.getUnknownDocs());
} }
RankEvalResponse response = new RankEvalResponse(); RankEvalResponse response = new RankEvalResponse();
// TODO move averaging to actual metric, also add other statistics
RankEvalResult result = new RankEvalResult(qualityTask.getTaskId(), qualitySum / specifications.size(), unknownDocs); RankEvalResult result = new RankEvalResult(qualityTask.getTaskId(), qualitySum / specifications.size(), unknownDocs);
response.setRankEvalResult(result); response.setRankEvalResult(result);
listener.onResponse(response); listener.onResponse(response);

View File

@ -35,6 +35,7 @@ import org.junit.AfterClass;
import org.junit.BeforeClass; import org.junit.BeforeClass;
import java.io.IOException; import java.io.IOException;
import java.util.ArrayList;
import java.util.List; import java.util.List;
public class QuerySpecTests extends ESTestCase { public class QuerySpecTests extends ESTestCase {
@ -50,7 +51,7 @@ public class QuerySpecTests extends ESTestCase {
@BeforeClass @BeforeClass
public static void init() throws IOException { public static void init() throws IOException {
aggsParsers = new AggregatorParsers(new ParseFieldRegistry<>("aggregation"), new ParseFieldRegistry<>("aggregation_pipes")); aggsParsers = new AggregatorParsers(new ParseFieldRegistry<>("aggregation"), new ParseFieldRegistry<>("aggregation_pipes"));
searchModule = new SearchModule(Settings.EMPTY, new NamedWriteableRegistry(), false); searchModule = new SearchModule(Settings.EMPTY, new NamedWriteableRegistry(), false, new ArrayList<>());
queriesRegistry = searchModule.getQueryParserRegistry(); queriesRegistry = searchModule.getQueryParserRegistry();
suggesters = searchModule.getSuggesters(); suggesters = searchModule.getSuggesters();
} }
@ -63,6 +64,7 @@ public class QuerySpecTests extends ESTestCase {
aggsParsers = null; aggsParsers = null;
} }
// TODO add some sort of roundtrip testing like we have now for queries?
public void testParseFromXContent() throws IOException { public void testParseFromXContent() throws IOException {
String querySpecString = " {\n" String querySpecString = " {\n"
+ " \"id\": \"my_qa_query\",\n" + " \"id\": \"my_qa_query\",\n"