Emulators

July 5, 2024 ยท View on GitHub

This part of scikit-learn_bench contains emulators - sklearn-like estimators wrapping other non-compliant frameworks' APIs.

Emulators are specified in configs using full module path and emulator name, for example:

{ "library": "sklbench.emulators.svs", "estimator": "NearestNeighbors" }

Emulators list

LibraryEmulator nameSupported methodsWrapped entity
FaissNearestNeighborsfit, kneighborsFlatL2, IVFFlat and IVFPQ index search. Supports both cpu and gpu devices.
RAFTNearestNeighborsfit, kneighborsFlatL2, IVFFlat, IVFPQ and CAGRA index search.
SVSNearestNeighborsfit, kneighborsVamana index search.

Documentation tree