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.

Field Summary

Fields

Modifier and Type	Field and Description
protected static Comparator<ScoreDoc>	scoreComparator

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)
protected static void	logSingleHit(IndexSearcher indexSearcher, LTRScoringQuery.ModelWeight modelWeight, int docid, LTRScoringQuery scoringQuery) Call this method if the scoreSingleHit(int, int, int, ScoreDoc, int, org.apache.solr.ltr.LTRScoringQuery.ModelWeight.ModelScorer, ScoreDoc[]) method indicated that the document's feature info should be logged.
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) Deprecated. Use scoreSingleHit(int, int, int, ScoreDoc, int, org.apache.solr.ltr.LTRScoringQuery.ModelWeight.ModelScorer, ScoreDoc[]) and logSingleHit(IndexSearcher, org.apache.solr.ltr.LTRScoringQuery.ModelWeight, int, LTRScoringQuery) instead.
protected static boolean	scoreSingleHit(int topN, int docBase, int hitUpto, ScoreDoc hit, int docID, LTRScoringQuery.ModelWeight.ModelScorer scorer, ScoreDoc[] reranked) Scores a single document and returns true if the document's feature info should be logged via the logSingleHit(IndexSearcher, org.apache.solr.ltr.LTRScoringQuery.ModelWeight, int, LTRScoringQuery) method.
protected static void	sortByScore(ScoreDoc[] reranked) Deprecated.

Methods inherited from class java.lang.Object

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

Field Detail

scoreComparator

protected static final Comparator<ScoreDoc> scoreComparator

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

@Deprecated
protected static void sortByScore(ScoreDoc[] reranked)

Deprecated.

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

@Deprecated
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

Deprecated. Use scoreSingleHit(int, int, int, ScoreDoc, int, org.apache.solr.ltr.LTRScoringQuery.ModelWeight.ModelScorer, ScoreDoc[]) and logSingleHit(IndexSearcher, org.apache.solr.ltr.LTRScoringQuery.ModelWeight, int, LTRScoringQuery) instead.

Throws:

IOException

logSingleHit

protected static void logSingleHit(IndexSearcher indexSearcher,
                                   LTRScoringQuery.ModelWeight modelWeight,
                                   int docid,
                                   LTRScoringQuery scoringQuery)

Call this method if the scoreSingleHit(int, int, int, ScoreDoc, int, org.apache.solr.ltr.LTRScoringQuery.ModelWeight.ModelScorer, ScoreDoc[]) method indicated that the document's feature info should be logged.

scoreSingleHit

protected static boolean scoreSingleHit(int topN,
                                        int docBase,
                                        int hitUpto,
                                        ScoreDoc hit,
                                        int docID,
                                        LTRScoringQuery.ModelWeight.ModelScorer scorer,
                                        ScoreDoc[] reranked)
                                 throws IOException

Scores a single document and returns true if the document's feature info should be logged via the logSingleHit(IndexSearcher, org.apache.solr.ltr.LTRScoringQuery.ModelWeight, int, LTRScoringQuery) method. Feature info logging is only necessary for the topN documents.

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