From 1891ba048e8ab4d37c18055e31f48c16c534cde8 Mon Sep 17 00:00:00 2001 From: arnaudon Date: Thu, 30 Nov 2023 11:45:49 +0100 Subject: [PATCH] fix --- MARBLE/preprocessing.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/MARBLE/preprocessing.py b/MARBLE/preprocessing.py index 1fbd5060..836106c4 100644 --- a/MARBLE/preprocessing.py +++ b/MARBLE/preprocessing.py @@ -113,7 +113,8 @@ def _compute_geometric_objects( Args: data: pytorch geometric data object - frac_geodesic_nb: fraction of geodesic neighbours relative to neighbours to fit the tangent spaces to + frac_geodesic_nb: fraction of geodesic neighbours relative to neighbours + to fit the tangent spaces to var_explained: fraction of variance explained by the local gauges local_gauges: whether to use local or global gauges