pyLucene - 如何使用BM25相似性而不是tf-idf

时间:2017-05-07 13:09:20

标签: lucene tf-idf pylucene

据我所知,pyLucene现在也提供BM25相似性。我正在使用pyLucene - 4.10.1,但无法找到关于如何使用BM25而不是tf-idf的任何示例。请指导。

1 个答案:

答案 0 :(得分:1)

尝试使用IndexSearity的IndexSarity来设置检索模型。

import lucene

from java.nio.file import Paths
from org.apache.lucene.store import SimpleFSDirectory
from org.apache.lucene.index import DirectoryReader
from org.apache.lucene.search import IndexSearcher
from org.apache.lucene.search.similarities import BM25Similarity


lucene.initVM(vmargs=['-Djava.awt.headless=true'])
directory = SimpleFSDirectory(Paths.get(INDEX_DIR))
searcher = IndexSearcher(DirectoryReader.open(directory))

searcher.setSimilarity(BM25Similarity())