core.datasets.retro.index

Exports:

• train_index: Train an index on representative vectors.

• add_to_index: Add vectors to a trained index.

• build_index: Wrapper function that calls above two functions.

Subpackages

• core.datasets.retro.index.indexes
  • core.datasets.retro.index.indexes.faiss_base
  • core.datasets.retro.index.indexes.faiss_par_add

Submodules

• core.datasets.retro.index.index
• core.datasets.retro.index.build
• core.datasets.retro.index.validate
• core.datasets.retro.index.factory
• core.datasets.retro.index.utils