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

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

Introduction

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

Prototype

public TrecTopicsReader() 

Source Link

Document

Constructor for Trec's TopicsReader

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   w w  w .  j a va 2 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: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);//  w w  w.  java 2  s  .c om

    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;/*www .  j  a  v  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   www.  j  av  a  2  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);//w ww.  j a v a2 s  . 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;
}