Skip to content

Commit e06e32c

Browse files
committed
minor changes
1 parent eae4282 commit e06e32c

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

ms2query/database/ann_vector_index.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def _umap_from_precomputed_knn(
8888
n_components: int = 2,
8989
min_dist: float = 0.1,
9090
spread: float = 1.0,
91-
random_state: int = 42,
91+
random_state: int = None,
9292
n_epochs: Optional[int] = None,
9393
negative_sample_rate: int = 5,
9494
verbose: bool = False,
@@ -512,11 +512,11 @@ def _normalize_query(self, query_fp) -> sp.csr_matrix:
512512
def compute_dr_coordinates(
513513
self,
514514
*,
515-
n_neighbors: int = 15,
515+
n_neighbors: int = 25,
516516
n_components: int = 2,
517-
min_dist: float = 0.1,
517+
min_dist: float = 0.2,
518518
spread: float = 1.0,
519-
random_state: int = 42,
519+
random_state: int = None,
520520
n_epochs: Optional[int] = None,
521521
negative_sample_rate: int = 5,
522522
ef: Optional[int] = None,
@@ -650,4 +650,4 @@ def _build_where_clause(where_sql: Optional[str]) -> str:
650650
def _count_rows(cursor: sqlite3.Cursor, table: str, where_clause: str) -> int:
651651
"""Count rows in table with optional WHERE clause."""
652652
cursor.execute(f"SELECT COUNT(1) FROM {table} {where_clause}")
653-
return int(cursor.fetchone()[0])
653+
return int(cursor.fetchone()[0])

0 commit comments

Comments
 (0)