Skip to content

Commit

Permalink
change default neighbors to 200 until rapidsai/cuml#6091 is addressed
Browse files Browse the repository at this point in the history
  • Loading branch information
thorstenwagner committed Oct 2, 2024
1 parent c96b92e commit 6798118
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tomotwin/modules/tools/umap.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ def create_parser(self, parentparser : ArgumentParser) -> ArgumentParser:
help='Previously fitted model')
parser.add_argument('-n', '--ncomponents', type=int, required=False, default=2,
help='Number of components')
parser.add_argument('--neighbors', type=int, required=False, default=200,
parser.add_argument('--neighbors', type=int, required=False, default=64,

Check warning on line 46 in tomotwin/modules/tools/umap.py

View check run for this annotation

Codecov / codecov/patch

tomotwin/modules/tools/umap.py#L46

Added line #L46 was not covered by tests
help='Previously fitted model')
parser.add_argument('--fit_sample_size', type=int, default=400000,
help='Sample size using for the fit of the umap')
Expand Down

0 comments on commit 6798118

Please sign in to comment.