9、自定义评分(lucene笔记)

这里我们是工程lucene_analyzer04

  • 1.自定义评分思路
    • 创建一个类继承于CustomScoreQuery
    • 覆盖里面的方法getCustomScoreProvider()
    • 创建一个类CustomScoreProvider
    • 覆盖CustomScoreProvider中的方法customScore(),在这个方法中创建评分规则。
  • 2.首先看一个示例
    我们在FileIndexUtil.java中定义一个随机数,这个随机数作为一个评分。
// 创建索引
public static void index(boolean hasNew) {
    IndexWriter writer = null;
    try {
        writer = new IndexWriter(directory, new IndexWriterConfig(
                Version.LUCENE_35, new StandardAnalyzer(Version.LUCENE_35)));
        if (hasNew) {
            writer.deleteAll();//如果我们要新建索引,那么将之前创建的删除
        }
        File file = new File("E:/myeclipse/Lucene/somefile");

        Document document = null;
        Random random = new Random();
        for (File f : file.listFiles()) {
            //将每个索引的评分设置成一个随机数
            int score = random.nextInt(600);
            document = new Document();
            document.add(new Field("content", new FileReader(f)));
            document.add(new Field("filename", f.getName(),
                    Field.Store.YES, Field.Index.NOT_ANALYZED));
            document.add(new Field("path", f.getAbsolutePath(),
                    Field.Store.YES, Field.Index.NOT_ANALYZED));
            document.add(new NumericField("date", Field.Store.YES, true)
                    .setLongValue(f.lastModified()));
            // 最后我们将字节数转换成kb
            document.add(new NumericField("size", Field.Store.YES, true)
                    .setIntValue((int) (f.length())));
            //这里我们使用score评分来创建索引,没有存储,搜索出来的时候为null
            //这里我自己随机设置的一个评分
            document.add(new NumericField("score", Field.Store.NO, true).setIntValue(score));
            writer.addDocument(document);
        }
    } catch (CorruptIndexException e) {
        e.printStackTrace();
    } catch (LockObtainFailedException e) {
        e.printStackTrace();
    } catch (IOException e) {
        e.printStackTrace();
    } finally {
        if (writer != null) {
            try {
                writer.close();
            } catch (CorruptIndexException e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

说明:定义随机评分之后再次生成索引。

  • 3.自定义评分规则
    MyScoreQuery.java
public void searchByScoreQuery(){
        try {
            IndexSearcher searcher = new IndexSearcher(IndexReader.open(FileIndexUtil.getDirectory()));
            Query q = new TermQuery(new Term("content", "json"));
            //1、创建一个评分域
            FieldScoreQuery fsq = new FieldScoreQuery("score", Type.INT);//如果是字符串选择BYTE类型,这里是int类型
            //2、根据评分域和原有的Query创建自定义的Query对象
            MyCustomScoreQuery query = new MyCustomScoreQuery(q, fsq);
            
            TopDocs tds = null;
            tds = searcher.search(query, 20);//使用自定义的query
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            for (ScoreDoc sd : tds.scoreDocs) {
                Document doc = searcher.doc(sd.doc);
                System.out.println("id:" + sd.doc + ",评分:" + sd.score 
                        + ",名称:" + doc.get("filename") + ",路径:" + doc.get("path")
                        + ",文件大小:" + doc.get("size") + ",日期:"
                        + sdf.format(new Date(Long.valueOf(doc.get("date")))));
            }
            searcher.close();
        } catch (CorruptIndexException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
private class MyCustomScoreQuery extends CustomScoreQuery{
        //subQuery表示原有的Query,valSrcQuery用于做评分的Query
        //CustomScoreQuery其实有很多构造方法,我们可以一次性传入很多评分规则
        public MyCustomScoreQuery(Query subQuery, ValueSourceQuery valSrcQuery) {
            super(subQuery, valSrcQuery);
        }
        
        //3、覆盖此方法
        @Override
        protected CustomScoreProvider getCustomScoreProvider(IndexReader reader)
                throws IOException {
            //默认情况实现的评分是通过   原有的评分*传入    进来的评分域得到最后的评分
            //而这种评分算法很显然是满足不了我们的要求的,于是我们自己为了根据不同的需求需要
            //自己进行评分设定
            /*
             * 自定义评分的步骤
             * (1)创建一个类继承于CustomScoreProvider
             * (2)覆盖CustomScoreProvider中的方法customScore()
             * */
            return new MyCustomScoreProvider(reader);
        }
    }
    
    private class MyCustomScoreProvider extends CustomScoreProvider{

        public MyCustomScoreProvider(IndexReader reader) {
            super(reader);
        }
        
        //此方法中subQueryScore表示默认文档的打分,valSrcScore表示的是评分域的打分
        @Override
        public float customScore(int doc, float subQueryScore, float valSrcScore)
                throws IOException {

            //这是按评分的降序排,之前默认的是subQueryScore * valSrcScore
            return subQueryScore / valSrcScore;
        }
    }

说明:

  • 1.这里我们是利用之前创建的那个随机评分,首先当然是创建评分域FieldScoreQuery ,这里我们选择的是score域。然后是根据评分域和原有的Query创建自定义的Query对象。
    自定义的Query对象MyCustomScoreQuery 必须继承CustomScoreQuery对象,然后需要一个构造方法,同时覆写getCustomScoreProvider方法,当然此方法中我们需要自定义评分规则类MyCustomScoreProvider,此类需要继承CustomScoreProvider,也是需要一个构造方法,同时覆写customScore方法,而我们真正的自定义评分规则就在这个方法中定义。
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,651评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,468评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,931评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,218评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,234评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,198评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,084评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,926评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,341评论 1 311
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,563评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,731评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,430评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,036评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,676评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,829评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,743评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,629评论 2 354

推荐阅读更多精彩内容

  • 1. Java基础部分 基础部分的顺序:基本语法,类相关的语法,内部类的语法,继承相关的语法,异常的语法,线程的语...
    子非鱼_t_阅读 31,623评论 18 399
  • 构建索引过程 文档是Lucene索引和被搜索的最小单位,一个文档包含一个或者多个域,而域则包含了真正 被搜索 的内...
    RagPanda阅读 3,481评论 0 2
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,652评论 18 139
  • 又是一年清明日, 山花烂漫踏青时。 稚子湖边嘻戏水, 不负春光好时节。
    Ekin_huang阅读 195评论 0 0
  • Reality is merely an illusion , albeit a very persisten...
    契德千字阅读 151评论 0 0