From 67981189c4205b7ab71b05b99b63fa499a001356 Mon Sep 17 00:00:00 2001 From: thorstenwagner Date: Wed, 2 Oct 2024 13:41:26 +0200 Subject: [PATCH] change default neighbors to 200 until https://github.com/rapidsai/cuml/issues/6091 is addressed --- tomotwin/modules/tools/umap.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tomotwin/modules/tools/umap.py b/tomotwin/modules/tools/umap.py index 0e83252..3aca702 100644 --- a/tomotwin/modules/tools/umap.py +++ b/tomotwin/modules/tools/umap.py @@ -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, help='Previously fitted model') parser.add_argument('--fit_sample_size', type=int, default=400000, help='Sample size using for the fit of the umap')