Package org.apache.solr.ltr

Class LTRRescorer

java.lang.Object

org.apache.lucene.search.Rescorer

org.apache.solr.ltr.LTRRescorer

public class LTRRescorer
extends org.apache.lucene.search.Rescorer

Implements the rescoring logic. The top documents returned by solr with their original scores, will be processed by a LTRScoringQuery that will assign a new score to each document. The top documents will be resorted based on the new score.

Constructor Summary

Constructors

Constructor	Description
LTRRescorer(LTRScoringQuery scoringQuery)

Method Summary

Modifier and Type	Method	Description
org.apache.lucene.search.Explanation	explain(org.apache.lucene.search.IndexSearcher searcher, org.apache.lucene.search.Explanation firstPassExplanation, int docID)
static LTRScoringQuery.FeatureInfo[]	extractFeaturesInfo(LTRScoringQuery.ModelWeight modelWeight, int docid, Float originalDocScore, List<org.apache.lucene.index.LeafReaderContext> leafContexts)
org.apache.lucene.search.TopDocs	rescore(org.apache.lucene.search.IndexSearcher searcher, org.apache.lucene.search.TopDocs firstPassTopDocs, int topN)	rescores the documents:
void	scoreFeatures(org.apache.lucene.search.IndexSearcher indexSearcher, org.apache.lucene.search.TopDocs firstPassTopDocs, int topN, LTRScoringQuery.ModelWeight modelWeight, org.apache.lucene.search.ScoreDoc[] hits, List<org.apache.lucene.index.LeafReaderContext> leaves, org.apache.lucene.search.ScoreDoc[] reranked)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LTRRescorer

public LTRRescorer(LTRScoringQuery scoringQuery)

Method Detail

rescore

public org.apache.lucene.search.TopDocs rescore(org.apache.lucene.search.IndexSearcher searcher,
                                                org.apache.lucene.search.TopDocs firstPassTopDocs,
                                                int topN)
                                         throws IOException

rescores the documents:

Specified by:

rescore in class org.apache.lucene.search.Rescorer

Parameters:

searcher - current IndexSearcher

firstPassTopDocs - documents to rerank;

topN - documents to return;

Throws:

IOException

scoreFeatures

public void scoreFeatures(org.apache.lucene.search.IndexSearcher indexSearcher,
                          org.apache.lucene.search.TopDocs firstPassTopDocs,
                          int topN,
                          LTRScoringQuery.ModelWeight modelWeight,
                          org.apache.lucene.search.ScoreDoc[] hits,
                          List<org.apache.lucene.index.LeafReaderContext> leaves,
                          org.apache.lucene.search.ScoreDoc[] reranked)
                   throws IOException

Throws:

IOException

explain

public org.apache.lucene.search.Explanation explain(org.apache.lucene.search.IndexSearcher searcher,
                                                    org.apache.lucene.search.Explanation firstPassExplanation,
                                                    int docID)
                                             throws IOException

Specified by:

explain in class org.apache.lucene.search.Rescorer

Throws:

IOException

extractFeaturesInfo

public static LTRScoringQuery.FeatureInfo[] extractFeaturesInfo(LTRScoringQuery.ModelWeight modelWeight,
                                                                int docid,
                                                                Float originalDocScore,
                                                                List<org.apache.lucene.index.LeafReaderContext> leafContexts)
                                                         throws IOException

Throws:

IOException