昇腾社区首页
中文
注册

create_reranker

功能描述

构造reranker。调用LocalReranker和TEIReranker的静态方法create返回实例。

函数原型

@classmethod
def create_reranker(cls, **kwargs):

输入参数说明

参数名

数据类型

可选/必选

说明

similarity_type

str

必选

该参数在kwargs中,reranker的类型。

可取值:

  • local_reranker
  • tei_reranker

**kwargs

Any

可选

除去similarity_type,其余参数为构造reranker的参数。

  • 如果是local_reranker,请参见类功能
  • 如果是tei_reranker,请参见类功能