org.apache.solr.ltr

Class LTRRescorer

java.lang.Object

org.apache.lucene.search.Rescorer

org.apache.solr.ltr.LTRRescorer

Direct Known Subclasses:

LTRInterleavingRescorer

public class LTRRescorer
extends 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 and Description
LTRRescorer()
LTRRescorer(LTRScoringQuery scoringQuery)

Method Summary

Modifier and Type	Method and Description
Explanation	explain(IndexSearcher searcher, Explanation firstPassExplanation, int docID)
static LTRScoringQuery.FeatureInfo[]	extractFeaturesInfo(LTRScoringQuery.ModelWeight modelWeight, int docid, Float originalDocScore, List<LeafReaderContext> leafContexts)
protected static Explanation	getExplanation(IndexSearcher searcher, int docID, LTRScoringQuery rerankingQuery)
protected static ScoreDoc[]	getFirstPassDocsRanked(TopDocs firstPassTopDocs)
protected static void	heapAdjust(ScoreDoc[] hits, int size, int root)
protected static void	heapify(ScoreDoc[] hits, int size)
TopDocs	rescore(IndexSearcher searcher, TopDocs firstPassTopDocs, int topN) rescores the documents:
void	scoreFeatures(IndexSearcher indexSearcher, int topN, LTRScoringQuery.ModelWeight modelWeight, ScoreDoc[] hits, List<LeafReaderContext> leaves, ScoreDoc[] reranked)
protected static void	scoreSingleHit(IndexSearcher indexSearcher, int topN, LTRScoringQuery.ModelWeight modelWeight, int docBase, int hitUpto, ScoreDoc hit, int docID, LTRScoringQuery rerankingQuery, LTRScoringQuery.ModelWeight.ModelScorer scorer, ScoreDoc[] reranked)
protected static void	sortByScore(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()

LTRRescorer

public LTRRescorer(LTRScoringQuery scoringQuery)

Method Detail

heapAdjust

protected static void heapAdjust(ScoreDoc[] hits,
                                 int size,
                                 int root)

heapify

protected static void heapify(ScoreDoc[] hits,
                              int size)

rescore

public TopDocs rescore(IndexSearcher searcher,
                       TopDocs firstPassTopDocs,
                       int topN)
                throws IOException

rescores the documents:

Specified by:

rescore in class Rescorer

Parameters:

searcher - current IndexSearcher

firstPassTopDocs - documents to rerank;

topN - documents to return;

Throws:

IOException

sortByScore

protected static void sortByScore(ScoreDoc[] reranked)

getFirstPassDocsRanked

protected static ScoreDoc[] getFirstPassDocsRanked(TopDocs firstPassTopDocs)

scoreFeatures

public void scoreFeatures(IndexSearcher indexSearcher,
                          int topN,
                          LTRScoringQuery.ModelWeight modelWeight,
                          ScoreDoc[] hits,
                          List<LeafReaderContext> leaves,
                          ScoreDoc[] reranked)
                   throws IOException

Throws:

IOException

scoreSingleHit

protected static void scoreSingleHit(IndexSearcher indexSearcher,
                                     int topN,
                                     LTRScoringQuery.ModelWeight modelWeight,
                                     int docBase,
                                     int hitUpto,
                                     ScoreDoc hit,
                                     int docID,
                                     LTRScoringQuery rerankingQuery,
                                     LTRScoringQuery.ModelWeight.ModelScorer scorer,
                                     ScoreDoc[] reranked)
                              throws IOException

Throws:

IOException

explain

public Explanation explain(IndexSearcher searcher,
                           Explanation firstPassExplanation,
                           int docID)
                    throws IOException

Specified by:

explain in class Rescorer

Throws:

IOException

getExplanation

protected static Explanation getExplanation(IndexSearcher searcher,
                                            int docID,
                                            LTRScoringQuery rerankingQuery)
                                     throws IOException

Throws:

IOException

extractFeaturesInfo

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

Throws:

IOException