Code

Add BGE-M3 Encoder#22

Code

Merged

seanmacavaney merged 21 commits intoterrierteam:masterterrierteam/pyterrier_dr:masterfrom andreaschari:masterandreaschari/pyterrier_dr:masterCopy head branch name to clipboardNov 22, 2024

+258-9

Commits

Commits on Apr 15, 2024

• add encoder for bge-m3 embeddings

  andreaschari

  committed
  4c479be

• fix return of _BGEM3_encode and return np array directly

  andreaschari

  committed
  d711650

Commits on Apr 22, 2024

• move encode arguments to BGEM3Encoder attributes

  andreaschari

  committed
  b94588f

Commits on Apr 29, 2024

• add refactor

  andreaschari

  committed
  c5d2763

• fix some typos + add ability to return token text for sparse encoder

  andreaschari

  committed
  c61134a

Commits on May 7, 2024

• remove multiple encoder class; merge into single class

  andreaschari

  committed
  7f60b2c

Commits on Aug 30, 2024

• Merge pull request #1 from andreaschari/refactor

  Show description for 8f38d3b
  andreaschari

  authored
  8f38d3b

• Merge branch 'terrierteam:master' into master

  andreaschari

  authored
  8015769

Commits on Nov 14, 2024

• Update README.md

  andreaschari

  authored
  4fc5e47

• Merge branch 'master' into master

  andreaschari

  authored
  8cb10ec

• remove "Factory" suffix

  andreaschari

  committed
  8b5ecbf

Commits on Nov 15, 2024

• update README

  andreaschari

  committed
  39c58e3

• refactor BGE-M3 implementation

  andreaschari

  committed
  fbe8a3e

• fix message

  andreaschari

  committed
  24bcaa8

Commits on Nov 18, 2024

• update readme

  andreaschari

  committed
  4dbe57e

• add bgem3 dependency

  andreaschari

  committed
  fbef79a

• refactor sparse transform output to dict; add encode_queries & encode_docs support

  andreaschari

  committed
  4750749

• refactor to handle empty inp; fix sparse output bug; rename multi-vector output column

  andreaschari

  committed
  e0e479b

• add tests for bgem3

  andreaschari

  committed
  0ebcde5

Commits on Nov 19, 2024

• add FlagEmbedding to test bge model

  cmacdonald

  authored
  701fe54

• update bgem-m3 multi-vec columns and README

  andreaschari

  committed
  fee706a