search

meta-soul / MetaSpore

A unified end-to-end machine intelligence platform
Apache License 2.0
642 stars 118 forks source link

[training] Two tower retrieval supports milvus #204

Closed codingfun2022 closed 2 years ago

codingfun2022 commented 2 years ago

Close #193

  1. Add TwoTowerIndexBuilder abstraction
  2. Implement TwoTowerFaissIndexBuilder for Faiss
  3. Implement TwoTowerMilvusIndexBuilder for Milvus
  4. Update TwoTowerRetrievalEstimator to use TwoTowerIndexBuilder
  5. Test: offline retrieval list matches that of online