Example usage for org.apache.lucene.benchmark.quality.trec TrecTopicsReader readQueries

List of usage examples for org.apache.lucene.benchmark.quality.trec TrecTopicsReader readQueries

Introduction

In this page you can find the example usage for org.apache.lucene.benchmark.quality.trec TrecTopicsReader readQueries.

Prototype

public QualityQuery[] readQueries(BufferedReader reader) throws IOException 

Source Link

Document

Read quality queries from trec format topics file.

Usage

From source file:com.mathworks.xzheng.benchmark.PrecisionRecall.java

License:Apache License

public static void main(String[] args) throws Throwable {

    File topicsFile = new File("src/lia/benchmark/topics.txt");
    File qrelsFile = new File("src/lia/benchmark/qrels.txt");
    Directory dir = FSDirectory.open(new File("indexes/MeetLucene"));

    IndexSearcher searcher = new IndexSearcher(DirectoryReader.open(dir));

    String docNameField = "filename";

    PrintWriter logger = new PrintWriter(System.out, true);

    TrecTopicsReader qReader = new TrecTopicsReader(); //#1
    QualityQuery qqs[] = qReader.readQueries( //#1
            new BufferedReader(new FileReader(topicsFile))); //#1

    Judge judge = new TrecJudge(new BufferedReader( //#2
            new FileReader(qrelsFile))); //#2

    judge.validateData(qqs, logger); //#3

    QualityQueryParser qqParser = new SimpleQQParser("title", "contents"); //#4

    QualityBenchmark qrun = new QualityBenchmark(qqs, qqParser, searcher, docNameField);
    SubmissionReport submitLog = null;/*from www. j av  a 2s. c om*/
    QualityStats stats[] = qrun.execute(judge, //#5
            submitLog, logger);

    QualityStats avg = QualityStats.average(stats); //#6
    avg.log("SUMMARY", 2, logger, "  ");
    dir.close();
}

From source file:io.datalayer.lucene.precision.PrecisionRecallMain.java

License:Apache License

public static void main(String[] args) throws Throwable {

    File topicsFile = new File("aos/lucene/precision/topics.txt");
    File qrelsFile = new File("aos/lucene/precision/qrels.txt");
    Directory dir = FSDirectory.open(new File("indexes/MeetLucene"));
    IndexReader reader = DirectoryReader.open(dir);
    IndexSearcher searcher = new IndexSearcher(reader);

    String docNameField = "filename";

    PrintWriter LOGGER = new PrintWriter(System.out, true);

    TrecTopicsReader qReader = new TrecTopicsReader();
    QualityQuery qqs[] = qReader.readQueries(new BufferedReader(new FileReader(topicsFile)));

    Judge judge = new TrecJudge(new BufferedReader(new FileReader(qrelsFile)));

    judge.validateData(qqs, LOGGER);//from   ww  w  . j  a v a 2s. c o  m

    QualityQueryParser qqParser = new SimpleQQParser("title", "contents");

    QualityBenchmark qrun = new QualityBenchmark(qqs, qqParser, searcher, docNameField);
    SubmissionReport submitLog = null;
    QualityStats stats[] = qrun.execute(judge, submitLog, LOGGER);

    QualityStats avg = QualityStats.average(stats);
    avg.log("SUMMARY", 2, LOGGER, "  ");

    dir.close();

}

From source file:lia.benchmark.PrecisionRecall.java

License:Apache License

public static void main(String[] args) throws Throwable {

    File topicsFile = new File("src/lia/benchmark/topics.txt");
    File qrelsFile = new File("src/lia/benchmark/qrels.txt");
    Directory dir = FSDirectory.open(new File("indexes/MeetLucene"));
    Searcher searcher = new IndexSearcher(dir, true);

    String docNameField = "filename";

    PrintWriter logger = new PrintWriter(System.out, true);

    TrecTopicsReader qReader = new TrecTopicsReader(); //#1
    QualityQuery qqs[] = qReader.readQueries( //#1
            new BufferedReader(new FileReader(topicsFile))); //#1

    Judge judge = new TrecJudge(new BufferedReader( //#2
            new FileReader(qrelsFile))); //#2

    judge.validateData(qqs, logger); //#3

    QualityQueryParser qqParser = new SimpleQQParser("title", "contents"); //#4

    QualityBenchmark qrun = new QualityBenchmark(qqs, qqParser, searcher, docNameField);
    SubmissionReport submitLog = null;// w  ww  .jav a2  s.  c  o  m
    QualityStats stats[] = qrun.execute(judge, //#5
            submitLog, logger);

    QualityStats avg = QualityStats.average(stats); //#6
    avg.log("SUMMARY", 2, logger, "  ");
    dir.close();
}

From source file:precisionRecall.precision.java

public static void main(String[] args) throws Throwable {

    Directory dir = FSDirectory.open(Paths.get(INDEX_DIR));

    File topicsFile = new File(
            "C:\\Users\\piyush\\Documents\\NetBeansProjects\\luceneFinal\\indexing\\topics.txt");

    File qrelsFile = new File(
            "C:\\Users\\piyush\\Documents\\NetBeansProjects\\luceneFinal\\indexing\\qrels.txt");

    IndexReader reader = DirectoryReader.open(dir);

    IndexSearcher searcher = new IndexSearcher(reader);

    String docNameField = "name";

    PrintWriter logger = new PrintWriter(System.out, true);

    TrecTopicsReader qReader = new TrecTopicsReader(); //#1

    QualityQuery qqs[] = qReader.readQueries( //#1

            new BufferedReader(new FileReader(topicsFile))); //#1

    Judge judge = new TrecJudge(new BufferedReader( //#2

            new FileReader(qrelsFile))); //#2

    judge.validateData(qqs, logger); //#3

    QualityQueryParser qqParser = new SimpleQQParser("title", "contents"); //#4

    QualityBenchmark qrun = new QualityBenchmark(qqs, qqParser, searcher, docNameField);

    SubmissionReport submitLog = null;/*from w  w  w  . j  a  v  a2 s.  co m*/

    QualityStats stats[] = qrun.execute(judge, //#5

            submitLog, logger);

    QualityStats avg = QualityStats.average(stats); //#6

    avg.log("SUMMARY", 2, logger, "  ");

    dir.close();

}

From source file:PrecisionRecall.PrecisionRecallMain.java

public BeanPrecisionRecall caculate() throws Throwable {

    File topicsFile = new File(getTopicsFile());
    File qrelsFile = new File(getQrelsFile());
    Directory dir = FSDirectory.open(new File(getIndexPath()));

    IndexReader reader = DirectoryReader.open(dir);
    IndexSearcher searcher = new IndexSearcher(reader);
    String docNameField = "filename";

    PrintWriter logger = new PrintWriter(System.out, true);

    TrecTopicsReader qReader = new TrecTopicsReader();
    QualityQuery qqs[] = qReader.readQueries(new BufferedReader(new FileReader(topicsFile)));

    Judge judge = new TrecJudge(new BufferedReader(new FileReader(qrelsFile)));

    judge.validateData(qqs, logger);// www . j ava 2s  .  c o m

    QualityQueryParser qqParser = new SimpleQQParser("title", "contents");

    QualityBenchmark qrun = new QualityBenchmark(qqs, qqParser, searcher, docNameField);
    SubmissionReport submitLog = null;

    QualityStats stats[] = qrun.execute(judge, submitLog, logger);

    QualityStats avg = QualityStats.average(stats);

    BeanPrecisionRecall bpr = new BeanPrecisionRecall();

    bpr.setAveragePrecision(avg.getAvp());
    bpr.setMmr(avg.getMRR());
    bpr.setRecall(avg.getRecall());
    bpr.setSearchTime(avg.getSearchTime());

    dir.close();
    return bpr;
}