From 5fb62c8b0e9ac3be154c7a051c83e962ffb5080d Mon Sep 17 00:00:00 2001 From: timsainb Date: Fri, 11 Jun 2021 17:29:51 -0700 Subject: [PATCH] changed default value to switch to python vs tensorflow based iteration --- umap/parametric_umap.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/umap/parametric_umap.py b/umap/parametric_umap.py index d1a78706..ee0f2f88 100644 --- a/umap/parametric_umap.py +++ b/umap/parametric_umap.py @@ -931,9 +931,9 @@ def construct_edge_dataset( def gather_index(index): return X[index] - # if X is > 2Gb in size, we need to use a different, slower method for + # if X is > 512Mb in size, we need to use a different, slower method for # batching data. - gather_indices_in_python = True if X.nbytes * 1e-9 > 2 else False + gather_indices_in_python = True if X.nbytes * 1e-9 > 0.5 else False def gather_X(edge_to, edge_from): # gather data from indexes (edges) in either numpy of tf, depending on array size