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" }| Library | Emulator name | Supported methods | Wrapped entity |
|---|---|---|---|
| Faiss | NearestNeighbors | fit, kneighbors |
FlatL2, IVFFlat and IVFPQ index search. Supports both cpu and gpu devices. |
| RAFT | NearestNeighbors | fit, kneighbors |
FlatL2, IVFFlat, IVFPQ and CAGRA index search. |
| SVS | NearestNeighbors | fit, kneighbors |
Vamana index search. |