diff options
Diffstat (limited to 'src/python/gudhi/tensorflow/cubical_layer.py')
-rw-r--r-- | src/python/gudhi/tensorflow/cubical_layer.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/python/gudhi/tensorflow/cubical_layer.py b/src/python/gudhi/tensorflow/cubical_layer.py index d68c7556..3304e719 100644 --- a/src/python/gudhi/tensorflow/cubical_layer.py +++ b/src/python/gudhi/tensorflow/cubical_layer.py @@ -37,17 +37,17 @@ class CubicalLayer(tf.keras.layers.Layer): """ TensorFlow layer for computing the persistent homology of a cubical complex """ - def __init__(self, dimensions, min_persistence=None, homology_coeff_field=11, **kwargs): + def __init__(self, homology_dimensions, min_persistence=None, homology_coeff_field=11, **kwargs): """ Constructor for the CubicalLayer class Parameters: - dimensions (List[int]): homology dimensions + homology_dimensions (List[int]): list of homology dimensions min_persistence (List[float]): minimum distance-to-diagonal of the points in the output persistence diagrams (default None, in which case 0. is used for all dimensions) homology_coeff_field (int): homology field coefficient. Must be a prime number. Default value is 11. Max is 46337. """ super().__init__(dynamic=True, **kwargs) - self.dimensions = dimensions + self.dimensions = homology_dimensions self.min_persistence = min_persistence if min_persistence != None else [0.] * len(self.dimensions) self.hcf = homology_coeff_field assert len(self.min_persistence) == len(self.dimensions) |