Code

Add cross encoder support#1615

Code

Merged

HenryL27 merged 18 commits intoopensearch-project:mainopensearch-project/ml-commons:mainfrom HenryL27:cross-encoderHenryL27/ml-commons:cross-encoderCopy head branch name to clipboardDec 7, 2023

+932-4

Commits

Commits on Nov 29, 2023

• add text similarity inputs and function name

  Show description for 3e9526b
  HenryL27

  committed
  3e9526b

  Copy full SHA for 3e9526b

• add text similarity cross encoder model

  Show description for a840ae7
  HenryL27

  committed
  a840ae7

  Copy full SHA for a840ae7

• add text similarity unit tests

  Show description for 3734ac4
  HenryL27

  committed
  3734ac4

  Copy full SHA for 3734ac4

• add text similarity input unittests

  Show description for 2ffea39
  HenryL27

  committed
  2ffea39

  Copy full SHA for 2ffea39

• add text similarity dataset unittests

  Show description for a03e7ce
  HenryL27

  committed
  a03e7ce

  Copy full SHA for a03e7ce

• add function name annotation

  Show description for 15e81bc
  HenryL27

  committed
  15e81bc

  Copy full SHA for 15e81bc

• refactor API to use single query

  Show description for 8dce6a9
  HenryL27

  committed
  8dce6a9

  Copy full SHA for 8dce6a9

• omit private from class vars

  Show description for 7871d8c
  

  HenryL27

  and

  navneet1v

  committed
  7871d8c

  Copy full SHA for 7871d8c

• change output name from logits to similarity

  Show description for 26bcbcd
  HenryL27

  committed
  26bcbcd

  Copy full SHA for 26bcbcd

• hashify isDLModel

  Show description for c32cf9e
  HenryL27

  committed
  c32cf9e

  Copy full SHA for c32cf9e

• add error message for non-torchscript cross encoders

  Show description for 0fc9028
  HenryL27

  committed
  0fc9028

  Copy full SHA for 0fc9028

• allow onnx, actually.

  Show description for 4ea18fa
  HenryL27

  committed
  4ea18fa

  Copy full SHA for 4ea18fa

• apply spotless after rebase

  Show description for 9d4e083
  HenryL27

  committed
  9d4e083

  Copy full SHA for 9d4e083

Commits on Dec 4, 2023

• add unittest for new mlinput toXcontent clause

  Show description for 6d7b429
  HenryL27

  committed
  6d7b429

  Copy full SHA for 6d7b429

Commits on Dec 5, 2023

• static DLModels

  Show description for 97e10b1
  HenryL27

  committed
  97e10b1

  Copy full SHA for 97e10b1

Commits on Dec 6, 2023

• add tests and error message tweaks

  Show description for 09e0f91
  HenryL27

  committed
  09e0f91

  Copy full SHA for 09e0f91

• name test models w framework

  Show description for b3e90b5
  HenryL27

  committed
  b3e90b5

  Copy full SHA for b3e90b5

• change pt->torch_script

  Show description for be113ed
  HenryL27

  committed
  be113ed

  Copy full SHA for be113ed