sentence-transformers/all-MiniLM-L6-v2 的 ONNX 版本

这是一个 sentence-transformers 模型：它将句子和段落映射到 384 维的稠密向量空间，可用于聚类或语义搜索等任务。该模型的 ONNX 版本是为 Metarank 重排序器设计的，用于计算语义相似度。

查看 Metarank 主文档 了解如何配置。

简而言之：

- type: field_match
  name: title_query_match
  rankingField: ranking.query
  itemField: item.title
  distance: cos 
  method:
    type: bert 
    model: metarank/all-MiniLM-L6-v2

构建模型

$> pip install -r requirements.txt
$> python convert.py

============= Diagnostic Run torch.onnx.export version 2.0.0+cu117 =============
verbose: False, log level: Level.ERROR
======================= 0 NONE 0 NOTE 0 WARNING 0 ERROR ========================

许可证

Apache 2.0