本文整理了Java中org.elasticsearch.search.rescore.QueryRescorer类的一些代码示例，展示了QueryRescorer类的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台，是从一些精选项目中提取出来的代码，具有较强的参考意义，能在一定程度帮忙到你。QueryRescorer类的具体详情如下：
包路径：org.elasticsearch.search.rescore.QueryRescorer
类名称：QueryRescorer

QueryRescorer介绍

暂无

代码示例

代码示例来源：origin: org.elasticsearch/elasticsearch

@Override
public TopDocs rescore(TopDocs topDocs, IndexSearcher searcher, RescoreContext rescoreContext) throws IOException {
  assert rescoreContext != null;
  if (topDocs == null || topDocs.totalHits == 0 || topDocs.scoreDocs.length == 0) {
    return topDocs;
  }
  final QueryRescoreContext rescore = (QueryRescoreContext) rescoreContext;
  org.apache.lucene.search.Rescorer rescorer = new org.apache.lucene.search.QueryRescorer(rescore.query()) {
    @Override
    protected float combine(float firstPassScore, boolean secondPassMatches, float secondPassScore) {
      if (secondPassMatches) {
        return rescore.scoreMode.combine(firstPassScore * rescore.queryWeight(), secondPassScore * rescore.rescoreQueryWeight());
      }
      // TODO: shouldn't this be up to the ScoreMode?  I.e., we should just invoke ScoreMode.combine, passing 0.0f for the
      // secondary score?
      return firstPassScore * rescore.queryWeight();
    }
  };
  // First take top slice of incoming docs, to be rescored:
  TopDocs topNFirstPass = topN(topDocs, rescoreContext.getWindowSize());
  // Save doc IDs for which rescoring was applied to be used in score explanation
  Set<Integer> topNDocIDs = Collections.unmodifiableSet(
    Arrays.stream(topNFirstPass.scoreDocs).map(scoreDoc -> scoreDoc.doc).collect(toSet()));
  rescoreContext.setRescoredDocs(topNDocIDs);
  // Rescore them:
  TopDocs rescored = rescorer.rescore(searcher, topNFirstPass, rescoreContext.getWindowSize());
  // Splice back to non-topN hits and resort all of them:
  return combine(topDocs, rescored, (QueryRescoreContext) rescoreContext);
}

代码示例来源：origin: com.strapdata.elasticsearch/elasticsearch

@Override
public QueryRescoreContext build(QueryShardContext context) throws IOException {
  org.elasticsearch.search.rescore.QueryRescorer rescorer = new org.elasticsearch.search.rescore.QueryRescorer();
  QueryRescoreContext queryRescoreContext = new QueryRescoreContext(rescorer);
  queryRescoreContext.setQuery(QueryBuilder.rewriteQuery(this.queryBuilder, context).toQuery(context));
  queryRescoreContext.setQueryWeight(this.queryWeight);
  queryRescoreContext.setRescoreQueryWeight(this.rescoreQueryWeight);
  queryRescoreContext.setScoreMode(this.scoreMode);
  if (this.windowSize != null) {
    queryRescoreContext.setWindowSize(this.windowSize);
  }
  return queryRescoreContext;
}

代码示例来源：origin: apache/servicemix-bundles

@Override
public TopDocs rescore(TopDocs topDocs, IndexSearcher searcher, RescoreContext rescoreContext) throws IOException {
  assert rescoreContext != null;
  if (topDocs == null || topDocs.totalHits == 0 || topDocs.scoreDocs.length == 0) {
    return topDocs;
  }
  final QueryRescoreContext rescore = (QueryRescoreContext) rescoreContext;
  org.apache.lucene.search.Rescorer rescorer = new org.apache.lucene.search.QueryRescorer(rescore.query()) {
    @Override
    protected float combine(float firstPassScore, boolean secondPassMatches, float secondPassScore) {
      if (secondPassMatches) {
        return rescore.scoreMode.combine(firstPassScore * rescore.queryWeight(), secondPassScore * rescore.rescoreQueryWeight());
      }
      // TODO: shouldn't this be up to the ScoreMode?  I.e., we should just invoke ScoreMode.combine, passing 0.0f for the
      // secondary score?
      return firstPassScore * rescore.queryWeight();
    }
  };
  // First take top slice of incoming docs, to be rescored:
  TopDocs topNFirstPass = topN(topDocs, rescoreContext.getWindowSize());
  // Save doc IDs for which rescoring was applied to be used in score explanation
  Set<Integer> topNDocIDs = Collections.unmodifiableSet(
    Arrays.stream(topNFirstPass.scoreDocs).map(scoreDoc -> scoreDoc.doc).collect(toSet()));
  rescoreContext.setRescoredDocs(topNDocIDs);
  // Rescore them:
  TopDocs rescored = rescorer.rescore(searcher, topNFirstPass, rescoreContext.getWindowSize());
  // Splice back to non-topN hits and resort all of them:
  return combine(topDocs, rescored, (QueryRescoreContext) rescoreContext);
}

代码示例来源：origin: org.apache.servicemix.bundles/org.apache.servicemix.bundles.elasticsearch

@Override
public TopDocs rescore(TopDocs topDocs, IndexSearcher searcher, RescoreContext rescoreContext) throws IOException {
  assert rescoreContext != null;
  if (topDocs == null || topDocs.totalHits == 0 || topDocs.scoreDocs.length == 0) {
    return topDocs;
  }
  final QueryRescoreContext rescore = (QueryRescoreContext) rescoreContext;
  org.apache.lucene.search.Rescorer rescorer = new org.apache.lucene.search.QueryRescorer(rescore.query()) {
    @Override
    protected float combine(float firstPassScore, boolean secondPassMatches, float secondPassScore) {
      if (secondPassMatches) {
        return rescore.scoreMode.combine(firstPassScore * rescore.queryWeight(), secondPassScore * rescore.rescoreQueryWeight());
      }
      // TODO: shouldn't this be up to the ScoreMode?  I.e., we should just invoke ScoreMode.combine, passing 0.0f for the
      // secondary score?
      return firstPassScore * rescore.queryWeight();
    }
  };
  // First take top slice of incoming docs, to be rescored:
  TopDocs topNFirstPass = topN(topDocs, rescoreContext.getWindowSize());
  // Save doc IDs for which rescoring was applied to be used in score explanation
  Set<Integer> topNDocIDs = Collections.unmodifiableSet(
    Arrays.stream(topNFirstPass.scoreDocs).map(scoreDoc -> scoreDoc.doc).collect(toSet()));
  rescoreContext.setRescoredDocs(topNDocIDs);
  // Rescore them:
  TopDocs rescored = rescorer.rescore(searcher, topNFirstPass, rescoreContext.getWindowSize());
  // Splice back to non-topN hits and resort all of them:
  return combine(topDocs, rescored, (QueryRescoreContext) rescoreContext);
}

代码示例来源：origin: harbby/presto-connectors

@Override
public TopDocs rescore(TopDocs topDocs, SearchContext context, RescoreSearchContext rescoreContext) throws IOException {
  assert rescoreContext != null;
  if (topDocs == null || topDocs.totalHits == 0 || topDocs.scoreDocs.length == 0) {
    return topDocs;
  }
  final QueryRescoreContext rescore = (QueryRescoreContext) rescoreContext;
  org.apache.lucene.search.Rescorer rescorer = new org.apache.lucene.search.QueryRescorer(rescore.query()) {
    @Override
    protected float combine(float firstPassScore, boolean secondPassMatches, float secondPassScore) {
      if (secondPassMatches) {
        return rescore.scoreMode.combine(firstPassScore * rescore.queryWeight(), secondPassScore * rescore.rescoreQueryWeight());
      }
      // TODO: shouldn't this be up to the ScoreMode?  I.e., we should just invoke ScoreMode.combine, passing 0.0f for the
      // secondary score?
      return firstPassScore * rescore.queryWeight();
    }
  };
  // First take top slice of incoming docs, to be rescored:
  TopDocs topNFirstPass = topN(topDocs, rescoreContext.window());
  // Rescore them:
  TopDocs rescored = rescorer.rescore(context.searcher(), topNFirstPass, rescoreContext.window());
  // Splice back to non-topN hits and resort all of them:
  return combine(topDocs, rescored, (QueryRescoreContext) rescoreContext);
}

代码示例来源：origin: com.strapdata.elasticsearch/elasticsearch

@Override
public TopDocs rescore(TopDocs topDocs, SearchContext context, RescoreSearchContext rescoreContext) throws IOException {
  assert rescoreContext != null;
  if (topDocs == null || topDocs.totalHits == 0 || topDocs.scoreDocs.length == 0) {
    return topDocs;
  }
  final QueryRescoreContext rescore = (QueryRescoreContext) rescoreContext;
  org.apache.lucene.search.Rescorer rescorer = new org.apache.lucene.search.QueryRescorer(rescore.query()) {
    @Override
    protected float combine(float firstPassScore, boolean secondPassMatches, float secondPassScore) {
      if (secondPassMatches) {
        return rescore.scoreMode.combine(firstPassScore * rescore.queryWeight(), secondPassScore * rescore.rescoreQueryWeight());
      }
      // TODO: shouldn't this be up to the ScoreMode?  I.e., we should just invoke ScoreMode.combine, passing 0.0f for the
      // secondary score?
      return firstPassScore * rescore.queryWeight();
    }
  };
  // First take top slice of incoming docs, to be rescored:
  TopDocs topNFirstPass = topN(topDocs, rescoreContext.window());
  // Rescore them:
  TopDocs rescored = rescorer.rescore(context.searcher(), topNFirstPass, rescoreContext.window());
  // Splice back to non-topN hits and resort all of them:
  return combine(topDocs, rescored, (QueryRescoreContext) rescoreContext);
}