利用Boost影响Lucene查询结果的排序

2024-03-09 10:03:11

转自：http://catastiger.iteye.com/blog/803796

前提:不对结果做sort操作. 
   在搜索中,并不是所有的Document和Fields都是平等的.有些技术会要求到对其Doucment或者Fields的权值改变,默认值为:1.0F,以上需求都是通过改变Document的boost因子来改变的. 下面是通过lucene3.0,IKAnalyzer 
1.通过设置doc boost改变排序结果

  1. /**
  2. * 设置DOC boost 值影响查询排序结果
  3. * @throws Exception
  4. */
  5. public void testBoost1() throws Exception{
  6. System.out.println("设置DOC boost 值影响查询排序结果");
  7. RAMDirectory ramDir = new RAMDirectory();
  8. Analyzer analyzer = new IKAnalyzer();
  9. IndexWriter iw = new IndexWriter(ramDir, analyzer, true ,IndexWriter.MaxFieldLength.LIMITED);
  10. String[] nameList = { "you are my friend", "a are my wife", "I love you" };
  11. String[] addList = { "b", "you are my wife", "c" };
  12. String[] fileList = { "1", "2", "3" };
  13. for (int i = 0; i < nameList.length; i++){
  14. Document doc = new Document();
  15. doc.add(new Field("name", nameList[i], Field.Store.YES, Field.Index.ANALYZED));
  16. doc.add(new Field("file", fileList[i], Field.Store.YES, Field.Index.ANALYZED));
  17. doc.add(new Field("address", addList[i], Field.Store.YES, Field.Index.ANALYZED));
  18. if (i == 2) {
  19. doc.setBoost(2.0f);
  20. }
  21. //            这里设置了第三个文档优先级最高，所以在搜索出来的结果中，该文档排在最前
  22. iw.addDocument(doc);
  23. }
  24. iw.close();
  25. IndexSearcher _searcher = new IndexSearcher(ramDir);
  26. String[] fields =new String[]{"name","address"};
  27. Query query=IKQueryParser.parseMultiField(fields, "you");
  28. TopDocs topDocs = _searcher.search(query,_searcher.maxDoc());
  29. ScoreDoc[] hits = topDocs.scoreDocs;
  30. for (int i = 0; i < hits.length; i++) {
  31. Document doc = _searcher.doc(hits[i].doc);
  32. System.out.println("name:"+doc.get("name"));
  33. System.out.println("file:"+doc.get("file"));
  34. }
  35. _searcher.close();
  36. }

if (i == 2) { doc.setBoost(2.0f); }这样I love you 将先输出， 
2.通过设置query 影响排序

  1. /**
  2. * 设置query boost值影响排序结果,如果有排序sort，则完全按照sort结果进行
  3. * @throws Exception
  4. */
  5. public void testBoost2() throws Exception{
  6. System.out.println("设置query boost值影响排序结果");
  7. RAMDirectory ramDir = new RAMDirectory();
  8. Analyzer analyzer = new IKAnalyzer();
  9. IndexWriter iw = new IndexWriter(ramDir, analyzer, true ,IndexWriter.MaxFieldLength.LIMITED);
  10. String[] nameList = { "you are my friend", "a are my wife", "I love you" };
  11. String[] addList = { "b", "you are my wife", "c" };
  12. String[] fileList = { "1", "2", "3" };
  13. for (int i = 0; i < nameList.length; i++)
  14. {
  15. Document doc = new Document();
  16. doc.add(new Field("name", nameList[i], Field.Store.YES, Field.Index.ANALYZED));
  17. doc.add(new Field("file", fileList[i], Field.Store.YES, Field.Index.ANALYZED));
  18. doc.add(new Field("address", addList[i], Field.Store.YES, Field.Index.ANALYZED));
  19. iw.addDocument(doc);
  20. }
  21. iw.close();
  22. IndexSearcher _searcher = new IndexSearcher(ramDir);
  23. BooleanQuery bq = new BooleanQuery();
  24. QueryParser _parser = new QueryParser(Version.LUCENE_30,"name",analyzer);
  25. Query  _query = _parser.parse("you");
  26. _query.setBoost(2f);
  27. QueryParser _parser1 = new QueryParser(Version.LUCENE_30,"address",analyzer);
  28. Query  _query1 = _parser1.parse("you");
  29. _query1.setBoost(1f);
  30. bq.add(_query, BooleanClause.Occur.SHOULD);
  31. bq.add(_query1, BooleanClause.Occur.SHOULD);
  32. //
  33. //          for(int i=0;i<2;i++){
  34. //              QueryParser parser = new MultiFieldQueryParser(Version.LUCENE_30,new String[] {"name", "address" }, analyzer);
  35. //              Query q1 = parser.parse("you");
  36. //              bq.add(q1, BooleanClause.Occur.MUST);
  37. //          }
  38. //
  39. //         SortField[] sortFields = new SortField[1];
  40. //         SortField sortField = new SortField("file", SortField.INT, true);//false升序，true降序
  41. //         sortFields[0] = sortField;
  42. //         Sort sort = new Sort(sortFields);
  43. //         TopDocs topDocs = _searcher.search(bq,null,_searcher.maxDoc(),sort);
  44. //
  45. TopDocs topDocs = _searcher.search(bq,_searcher.maxDoc());
  46. ScoreDoc[] hits = topDocs.scoreDocs;
  47. for (int i = 0; i < hits.length; i++) {
  48. Document doc = _searcher.doc(hits[i].doc);
  49. System.out.println("name:"+doc.get("name"));
  50. System.out.println("file:"+doc.get("file"));
  51. }
  52. _searcher.close();
  53. }

结果如下：（name 的boost最高，所以name优先于address排序在前面） 
设置query boost值影响排序结果 
name:you are my friend 
file:1 
name:I love you 
file:3 
name:a are my wife 
file:2

3.通过设置fields 的boost 影响排序

  1. /**
  2. * 设置field boost 值影响查询排序结果,有排序则按照排序
  3. * @throws Exception
  4. */
  5. //没设置field boost 213 设置后是132
  6. public void testBoost3() throws Exception{
  7. System.out.println("设置fields boost 值影响查询排序结果");
  8. RAMDirectory ramDir = new RAMDirectory();
  9. Analyzer analyzer = new IKAnalyzer();
  10. IndexWriter iw = new IndexWriter(ramDir, analyzer, true ,IndexWriter.MaxFieldLength.LIMITED);
  11. String[] nameList = { "you are my friend", "a are my wife", "I love you" };
  12. String[] addList = { "b", "you are my wife", "c" };
  13. String[] fileList = { "1", "2", "3" };
  14. for (int i = 0; i < nameList.length; i++)
  15. {
  16. Document doc = new Document();
  17. Field nameField =  new Field("name", nameList[i], Field.Store.YES, Field.Index.ANALYZED);
  18. nameField.setBoost(20f);
  19. doc.add(nameField);
  20. doc.add(new Field("file", fileList[i], Field.Store.YES, Field.Index.ANALYZED));
  21. Field f = new Field("address", addList[i], Field.Store.YES, Field.Index.ANALYZED);
  22. f.setBoost(30f);
  23. doc.add(f);
  24. iw.addDocument(doc);
  25. }
  26. iw.close();
  27. IndexSearcher _searcher = new IndexSearcher(ramDir);
  28. String[] fields =new String[]{"name","file","address"};
  29. Query query=IKQueryParser.parseMultiField(fields, "you");
  30. //        SortField[] sortFields = new SortField[1];
  31. //        SortField sortField = new SortField("file", SortField.INT, true);//false升序，true降序
  32. //        sortFields[0] = sortField;
  33. //        Sort sort = new Sort(sortFields);
  34. //        TopDocs topDocs = _searcher.search(query,null,_searcher.maxDoc(),sort);
  35. TopDocs topDocs = _searcher.search(query,_searcher.maxDoc());
  36. ScoreDoc[] hits = topDocs.scoreDocs;
  37. for (int i = 0; i < hits.length; i++) {
  38. Document doc = _searcher.doc(hits[i].doc);
  39. System.out.println("name:"+doc.get("name"));
  40. System.out.println("file:"+doc.get("file"));
  41. }
  42. _searcher.close();
  43. }

结果如下：（address 的boost最高，先排在前面了） 
设置fields boost 值影响查询排序结果 
name:a are my wife 
file:2 
name:you are my friend 
file:1 
name:I love you 
file:3

上一篇：MongoDB-数据&权限管理(4)


下一篇：python读取文件操作.CSV