Tarang Jain
Tarang Jain
- [x] All points distance membership vector - [x] All points outlier membership vector - [x] All points probability in some cluster - [x] All points membership vector - [x]...
PR for HDBSCAN approximate_predict - [x] Building cluster_map - [x] Modifying PredictionData class - [ ] Obtaining nearest neighbor in MR space - [ ] Computing probability - [ ]...
- [x] Updated `randomforestclassifier.pyx` for accepting` numpy.random.RandomState`, `cupy.random.RandomState` and `numpy.random.Generator` objects in addition to accepting integers for the random_state variable. - [x] Add tests - [x] Update documentation
For computing the core distances, during training the (min_samples+1)-th neighbor is considered, but while building the PredictionData object, the (min_samples)-th neighbor is considered. The specific parts in the code that...
Benchmark CAGRA build + DiskANN (in memory) search on the graph.
CAGRA graph construction with IVF-PQ will support inner product with PR https://github.com/rapidsai/raft/pull/2260. The same needs to be done for NN Descent. This is not done currently because NN Descent does...
- [x] RMM pooled resource for RAFT enabled FAISS - [x] Update yaml config for `faiss_gpu_ivf_flat`, `faiss_gpu_ivf_pq` - [x] Small fix FAISS GPU IVFPQ params - [x] Update `get_faiss.cmake` -...
- Add InnerProduct metric for CAGRA-Q - Add cosine metric for CAGRA and CAGRA-Q
Remove the dependency on `raft::compiled` and modify GPU implementations to use cuVS backend in place of RAFT. A deeper insight into the dependency: FAISS gets the ANN algorithm implementations such...
Noticed an error after `raft-ann-bench` pkg is renamed to `raft_ann_bench`. ``` Traceback (most recent call last): File "", line 198, in _run_module_as_main File "", line 88, in _run_code File "/raid/tarangj/miniconda3/envs/all_cuda-122_arch-x86_64/lib/python3.11/site-packages/raft_ann_bench/run/__main__.py",...