Skip to content

Commit

Permalink
add buffer_size param to search centroid exhaustive
Browse files Browse the repository at this point in the history
  • Loading branch information
tony-kuo committed Sep 13, 2024
1 parent d2b8364 commit 5f2f3e9
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions src/scimilarity/cell_query.py
Original file line number Diff line number Diff line change
Expand Up @@ -648,6 +648,7 @@ def search_centroid_exhaustive(
max_dist: float = 0.03,
qc: bool = True,
qc_params: dict = {"k_clusters": 10},
buffer_size: int = 100000,
random_seed: int = 4,
) -> Tuple[
"numpy.ndarray",
Expand All @@ -673,6 +674,8 @@ def search_centroid_exhaustive(
qc_params: dict, default: {'k_clusters': 10}
Parameters for the QC:
k_clusters: the number of clusters in kmeans clustering
buffer_size: int, default: 100000
Batch size for processing cells.
random_seed: int, default: 1
Random seed for k-means clustering
Expand Down Expand Up @@ -709,6 +712,7 @@ def search_centroid_exhaustive(
nn_idxs, nn_dists, metadata = self.search_exhaustive(
centroid_embedding,
max_dist=max_dist,
buffer_size=buffer_size,
)

qc_stats = {}
Expand Down

0 comments on commit 5f2f3e9

Please sign in to comment.