org.apache.solr.ltr

Class LTRRescorer

java.lang.Object

org.apache.lucene.search.Rescorer

org.apache.solr.ltr.LTRRescorer

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(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)
TopDocs	rescore(IndexSearcher searcher, TopDocs firstPassTopDocs, int topN) rescores the documents:
void	scoreFeatures(IndexSearcher indexSearcher, TopDocs firstPassTopDocs, int topN, LTRScoringQuery.ModelWeight modelWeight, ScoreDoc[] hits, List<LeafReaderContext> leaves, 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 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

scoreFeatures

public void scoreFeatures(IndexSearcher indexSearcher,
                          TopDocs firstPassTopDocs,
                          int topN,
                          LTRScoringQuery.ModelWeight modelWeight,
                          ScoreDoc[] hits,
                          List<LeafReaderContext> leaves,
                          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

extractFeaturesInfo

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

Throws:

IOException