pyterrier_colbert 源码
皮特里尔-科尔伯特 高级绑定,包括密集的索引和检索。 用法 给定现有的ColBERT检查点,可以如下创建端到端的ColBERT密集检索索引: from pyterrier_colbert . indexing import ColBERTIndexer indexer = ColBERTIndexer ( "/path/to/checkpoint.dnn" , "/path/to/index" , "index_name" ) indexer . index ( dataset . get_corpus_iter ()) 端到端的ColBERT密集检索管道可以表述为: from pyterrier_colbert . ranking import ColBERTFactory pytcolbert = ColBERTFactory ( "/path/to/checkpoint.dnn
下载地址
用户评论