Skip to content

Commit fed0d5c

Browse files
authored
[auto-merge] branch-25.06 to branch-25.08 [skip ci] [bot] (#962)
auto-merge triggered by github actions on `branch-25.06` to create a PR keeping `branch-25.08` up-to-date. If this PR is unable to be merged due to conflicts, it will remain open until manually fix.
2 parents 7267705 + c316a70 commit fed0d5c

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

python/src/spark_rapids_ml/umap.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ def __init__(self) -> None:
366366
"build_kwds",
367367
(
368368
f"Build algorithm argument {{'nnd_graph_degree': 64, 'nnd_intermediate_graph_degree': 128, 'nnd_max_iterations': 20,"
369-
f" 'nnd_termination_threshold': 0.0001, 'nnd_return_distances': True, 'nnd_n_clusters': 1}} Note that nnd_n_clusters > 1"
369+
f" 'nnd_termination_threshold': 0.0001, 'nnd_n_clusters': 1, 'nnd_overlap_factor': 2}} Note that nnd_n_clusters > 1"
370370
f" will result in batch-building with NN Descent."
371371
),
372372
typeConverter=DictTypeConverters._toDict,
@@ -787,7 +787,7 @@ class UMAP(UMAPClass, _CumlEstimatorSupervised, _UMAPCumlParams):
787787
788788
build_kwds : dict (optional, default=None)
789789
Build algorithm argument {'nnd_graph_degree': 64, 'nnd_intermediate_graph_degree': 128, 'nnd_max_iterations': 20,
790-
'nnd_termination_threshold': 0.0001, 'nnd_return_distances': True, 'nnd_n_clusters': 1} Note that nnd_n_clusters > 1
790+
'nnd_termination_threshold': 0.0001, 'nnd_n_clusters': 1, 'nnd_overlap_factor': 2} Note that nnd_n_clusters > 1
791791
will result in batch-building with NN Descent.
792792
793793
sample_fraction : float (optional, default=1.0)

python/tests/test_umap.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -675,8 +675,8 @@ def test_umap_build_algo(gpu_number: int, metric: str) -> None:
675675
"nnd_intermediate_graph_degree": 128,
676676
"nnd_max_iterations": 40,
677677
"nnd_termination_threshold": 0.0001,
678-
"nnd_return_distances": True,
679678
"nnd_n_clusters": 5,
679+
"nnd_overlap_factor": 2,
680680
}
681681

682682
umap = UMAP(

0 commit comments

Comments
 (0)